Push model using huggingface_hub.

README.md

@@ -1,5 +1,6 @@
 ---
 license: apache-2.0
+library_name: transformers
 tags:
 - trl
 - ppo
@@ -25,7 +26,7 @@ You can then generate text as follows:
 ```python
 from transformers import pipeline
 
-generator = pipeline("text-generation", model="claudiubarbu//tmp/tmp57jg9o2i/claudiubarbu/HW2-ppo")
+generator = pipeline("text-generation", model="claudiubarbu//tmp/tmpdmfjw943/claudiubarbu/HW2-ppo")
 outputs = generator("Hello, my llama is cute")
 ```
 
@@ -35,8 +36,8 @@ If you want to use the model for training or to obtain the outputs from the valu
 from transformers import AutoTokenizer
 from trl import AutoModelForCausalLMWithValueHead
 
-tokenizer = AutoTokenizer.from_pretrained("claudiubarbu//tmp/tmp57jg9o2i/claudiubarbu/HW2-ppo")
-model = AutoModelForCausalLMWithValueHead.from_pretrained("claudiubarbu//tmp/tmp57jg9o2i/claudiubarbu/HW2-ppo")
+tokenizer = AutoTokenizer.from_pretrained("claudiubarbu//tmp/tmpdmfjw943/claudiubarbu/HW2-ppo")
+model = AutoModelForCausalLMWithValueHead.from_pretrained("claudiubarbu//tmp/tmpdmfjw943/claudiubarbu/HW2-ppo")
 
 inputs = tokenizer("Hello, my llama is cute", return_tensors="pt")
 outputs = model(**inputs, labels=inputs["input_ids"])

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:20385ac034306cbc59dbec1fc73d6952bb0c09f3846779e763d93b8b7a4a7693
+oid sha256:7c1d0de370ae0f121a8ff56e74e1eda0e35af6a6397e1075596486ca8533bc61
 size 497777468

tokenizer_config.json

@@ -13,8 +13,12 @@
   "bos_token": "<|endoftext|>",
   "clean_up_tokenization_spaces": true,
   "eos_token": "<|endoftext|>",
+  "max_length": 512,
   "model_max_length": 1024,
   "pad_token": "<|endoftext|>",
+  "stride": 0,
   "tokenizer_class": "GPT2Tokenizer",
+  "truncation_side": "right",
+  "truncation_strategy": "longest_first",
   "unk_token": "<|endoftext|>"
 }