arianaazarbal commited on
Commit
92b8207
·
verified ·
1 Parent(s): 3d56146

Push model using huggingface_hub.

Browse files
README.md CHANGED
@@ -26,7 +26,7 @@ You can then generate text as follows:
26
  ```python
27
  from transformers import pipeline
28
 
29
- generator = pipeline("text-generation", model="arianaazarbal//tmp/tmptpdt1_2u/arianaazarbal/ppo-finetuned-model")
30
  outputs = generator("Hello, my llama is cute")
31
  ```
32
 
@@ -36,8 +36,8 @@ If you want to use the model for training or to obtain the outputs from the valu
36
  from transformers import AutoTokenizer
37
  from trl import AutoModelForCausalLMWithValueHead
38
 
39
- tokenizer = AutoTokenizer.from_pretrained("arianaazarbal//tmp/tmptpdt1_2u/arianaazarbal/ppo-finetuned-model")
40
- model = AutoModelForCausalLMWithValueHead.from_pretrained("arianaazarbal//tmp/tmptpdt1_2u/arianaazarbal/ppo-finetuned-model")
41
 
42
  inputs = tokenizer("Hello, my llama is cute", return_tensors="pt")
43
  outputs = model(**inputs, labels=inputs["input_ids"])
 
26
  ```python
27
  from transformers import pipeline
28
 
29
+ generator = pipeline("text-generation", model="arianaazarbal//tmp/tmpu6dj7w01/arianaazarbal/ppo-finetuned-model")
30
  outputs = generator("Hello, my llama is cute")
31
  ```
32
 
 
36
  from transformers import AutoTokenizer
37
  from trl import AutoModelForCausalLMWithValueHead
38
 
39
+ tokenizer = AutoTokenizer.from_pretrained("arianaazarbal//tmp/tmpu6dj7w01/arianaazarbal/ppo-finetuned-model")
40
+ model = AutoModelForCausalLMWithValueHead.from_pretrained("arianaazarbal//tmp/tmpu6dj7w01/arianaazarbal/ppo-finetuned-model")
41
 
42
  inputs = tokenizer("Hello, my llama is cute", return_tensors="pt")
43
  outputs = model(**inputs, labels=inputs["input_ids"])
adapter_config.json CHANGED
@@ -24,10 +24,10 @@
24
  "rank_pattern": {},
25
  "revision": null,
26
  "target_modules": [
27
- "q_proj",
28
- "k_proj",
29
  "v_proj",
30
- "o_proj"
 
31
  ],
32
  "task_type": "CAUSAL_LM",
33
  "trainable_token_indices": null,
 
24
  "rank_pattern": {},
25
  "revision": null,
26
  "target_modules": [
27
+ "o_proj",
 
28
  "v_proj",
29
+ "q_proj",
30
+ "k_proj"
31
  ],
32
  "task_type": "CAUSAL_LM",
33
  "trainable_token_indices": null,
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f42b35ae1c092012df22d58de5dc83485c1141a224c35aacc9510e363f95d1ec
3
  size 40400200
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0deea2490a4e8cf6b88413eda4ef11940cc91705fb6d75968dbe880f41979bc
3
  size 40400200
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8972728555d3821281ff050a93bdca6d904c1cfb33b498f0f8e9138a42492af2
3
  size 16277
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1db72a3ab3141e17bd54bfb9fb31d38b29129500c4b1b57f42fdb659267dff1e
3
  size 16277