ArthurZ
/

opt-350m

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

ArthurZ HF staff commited on Jun 21, 2022

Commit

d99439a

•

1 Parent(s): 104e274

add model

Files changed (2) hide show

config.json +2 -2
pytorch_model.bin +2 -2

config.json CHANGED Viewed

@@ -3,7 +3,7 @@
   "activation_dropout": 0.0,
   "activation_function": "relu",
   "architectures": [
-    "OPTModel"
   ],
   "attention_dropout": 0.0,
   "bos_token_id": 2,
@@ -20,7 +20,7 @@
   "num_hidden_layers": 24,
   "pad_token_id": 1,
   "prefix": "</s>",
-  "torch_dtype": "float32",
   "transformers_version": "4.21.0.dev0",
   "use_cache": true,
   "vocab_size": 50272,

   "activation_dropout": 0.0,
   "activation_function": "relu",
   "architectures": [
+    "OPTForCausalLM"
   ],
   "attention_dropout": 0.0,
   "bos_token_id": 2,
   "num_hidden_layers": 24,
   "pad_token_id": 1,
   "prefix": "</s>",
+  "torch_dtype": "float16",
   "transformers_version": "4.21.0.dev0",
   "use_cache": true,
   "vocab_size": 50272,

pytorch_model.bin CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c450f836f017007a742704fc9f469acae8d94e2a6e785151016532c4b680f7ef
-size 1324906489

 version https://git-lfs.github.com/spec/v1
+oid sha256:6317faef853c1e43396f38b83e8f5cde544ce9847344e911ffd0d2bf720151ad
+size 662516025