Push model using huggingface_hub.

README.md

@@ -25,7 +25,7 @@ You can then generate text as follows:
 ```python
 from transformers import pipeline
 
-generator = pipeline("text-generation", model="damienbenveniste//var/folders/qj/lfvfq6590q5fn7hnwx6c29k80000gn/T/tmppue5zznd/damienbenveniste/mistral-ppo")
+generator = pipeline("text-generation", model="damienbenveniste//private/var/folders/dy/k5ycdcns28s2cxl8hc76v2mr0000gn/T/tmpl2a_v6sk/damienbenveniste/mistral-ppo")
 outputs = generator("Hello, my llama is cute")
 ```
 
@@ -35,8 +35,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("damienbenveniste//var/folders/qj/lfvfq6590q5fn7hnwx6c29k80000gn/T/tmppue5zznd/damienbenveniste/mistral-ppo")
-model = AutoModelForCausalLMWithValueHead.from_pretrained("damienbenveniste//var/folders/qj/lfvfq6590q5fn7hnwx6c29k80000gn/T/tmppue5zznd/damienbenveniste/mistral-ppo")
+tokenizer = AutoTokenizer.from_pretrained("damienbenveniste//private/var/folders/dy/k5ycdcns28s2cxl8hc76v2mr0000gn/T/tmpl2a_v6sk/damienbenveniste/mistral-ppo")
+model = AutoModelForCausalLMWithValueHead.from_pretrained("damienbenveniste//private/var/folders/dy/k5ycdcns28s2cxl8hc76v2mr0000gn/T/tmpl2a_v6sk/damienbenveniste/mistral-ppo")
 
 inputs = tokenizer("Hello, my llama is cute", return_tensors="pt")
 outputs = model(**inputs, labels=inputs["input_ids"])

config.json

@@ -3,8 +3,10 @@
   "architectures": [
     "MistralForCausalLM"
   ],
+  "attention_dropout": 0.0,
   "bos_token_id": 1,
   "eos_token_id": 2,
+  "head_dim": 48,
   "hidden_act": "silu",
   "hidden_size": 768,
   "initializer_range": 0.02,
@@ -19,7 +21,7 @@
   "sliding_window": 768,
   "tie_word_embeddings": false,
   "torch_dtype": "float32",
-  "transformers_version": "4.35.2",
+  "transformers_version": "4.44.2",
   "use_cache": true,
   "vocab_size": 32000
 }

generation_config.json

@@ -2,5 +2,5 @@
   "_from_model_config": true,
   "bos_token_id": 1,
   "eos_token_id": 2,
-  "transformers_version": "4.35.2"
+  "transformers_version": "4.44.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d8de2bb878800087d0c7fbc318e19ee4fc1d428a587daa2bab14ccc6363b42ae
+oid sha256:85ee50a22dabf1652e2365843c33fbb25cab5fa12acf25bc0c36147211f49b65
 size 338200972

tokenizer.json

@@ -31,23 +31,13 @@
       "special": true
     }
   ],
-  "normalizer": {
-    "type": "Sequence",
-    "normalizers": [
-      {
-        "type": "Prepend",
-        "prepend": "▁"
-      },
-      {
-        "type": "Replace",
-        "pattern": {
-          "String": " "
-        },
-        "content": "▁"
-      }
-    ]
+  "normalizer": null,
+  "pre_tokenizer": {
+    "type": "Metaspace",
+    "replacement": "▁",
+    "prepend_scheme": "first",
+    "split": false
   },
-  "pre_tokenizer": null,
   "post_processor": {
     "type": "TemplateProcessing",
     "single": [
@@ -134,6 +124,7 @@
     "end_of_word_suffix": null,
     "fuse_unk": true,
     "byte_fallback": true,
+    "ignore_merges": false,
     "vocab": {
       "<unk>": 0,
       "<s>": 1,

tokenizer_config.json

@@ -1,4 +1,7 @@
 {
+  "add_bos_token": true,
+  "add_eos_token": false,
+  "add_prefix_space": null,
   "added_tokens_decoder": {
     "0": {
       "content": "<unk>",
@@ -29,8 +32,8 @@
   "bos_token": "<s>",
   "clean_up_tokenization_spaces": false,
   "eos_token": "</s>",
-  "legacy": true,
-  "max_length": 30000,
+  "legacy": false,
+  "max_length": 512,
   "model_max_length": 1000000000000000019884624838656,
   "pad_to_multiple_of": null,
   "pad_token": "</s>",