stanrom
/

internlm-xcomposer-7b-4bit

Feature Extraction

InternLMXComposer

Model card Files Files and versions Community

stanrom commited on May 25

Commit

9a6894c

•

1 Parent(s): 52966eb

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -10,7 +10,7 @@
   },
   "bias": true,
   "bos_token_id": 1,
-  "device": "cpu",
   "eos_token_id": 2,
   "hidden_act": "silu",
   "hidden_size": 4096,

   },
   "bias": true,
   "bos_token_id": 1,
+  "device": "cuda:0",
   "eos_token_id": 2,
   "hidden_act": "silu",
   "hidden_size": 4096,