NbAiLab
/

nb-llama-3.1-70B

Text Generation

Model card Files Files and versions Community

Change dtype to bf16

#1

by saattrupdan - opened 26 days ago

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -29,7 +29,7 @@
   },
   "rope_theta": 500000.0,
   "tie_word_embeddings": false,
-  "torch_dtype": "float32",
   "transformers_version": "4.46.3",
   "use_cache": true,
   "vocab_size": 128256

   },
   "rope_theta": 500000.0,
   "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
   "transformers_version": "4.46.3",
   "use_cache": true,
   "vocab_size": 128256