OpenOpenAI
/

ppo_iter_14

Text Generation

text-generation-inference

Model card Files Files and versions

OpenOpenAI commited on Aug 31, 2024

Commit

3f245ca

·

verified ·

1 Parent(s): 358b000

Upload config

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,7 +1,7 @@
 {
   "_name_or_path": "/workspace/OpenRLHF3/checkpoint/llama-3-8b-rlhf/iter_14",
   "architectures": [
-    "PhiModel"
   ],
   "attention_dropout": 0.0,
   "bos_token_id": null,
@@ -23,7 +23,7 @@
   "rope_scaling": null,
   "rope_theta": 10000.0,
   "tie_word_embeddings": false,
-  "torch_dtype": "float32",
   "transformers_version": "4.43.1",
   "use_cache": false,
   "vocab_size": 51200

 {
   "_name_or_path": "/workspace/OpenRLHF3/checkpoint/llama-3-8b-rlhf/iter_14",
   "architectures": [
+    "PhiForCausalLM"
   ],
   "attention_dropout": 0.0,
   "bos_token_id": null,
   "rope_scaling": null,
   "rope_theta": 10000.0,
   "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
   "transformers_version": "4.43.1",
   "use_cache": false,
   "vocab_size": 51200