rachiteagles
/

cover_letter

Text Generation

text-generation-inference

Inference Endpoints

4-bit precision

Model card Files Files and versions Community

rachiteagles commited on Apr 27

Commit

2988eb6

•

1 Parent(s): c8182fd

Upload config

Files changed (1) hide show

config.json +1 -2

config.json CHANGED Viewed

@@ -21,8 +21,7 @@
   "pad_token_id": 0,
   "rms_norm_eps": 1e-06,
   "rope_theta": 10000.0,
-  "task_name": "text-generation",
-  "torch_dtype": "float16",
   "transformers_version": "4.40.0",
   "use_cache": true,
   "vocab_size": 256000

   "pad_token_id": 0,
   "rms_norm_eps": 1e-06,
   "rope_theta": 10000.0,
+  "torch_dtype": "bfloat16",
   "transformers_version": "4.40.0",
   "use_cache": true,
   "vocab_size": 256000