stanrom
/

internlm-xcomposer-7b-4bit

stanrom commited on May 25

Commit

4caf6c7

•

1 Parent(s): 9a6894c

Update configuration_InternLM_XComposer.py

Files changed (1) hide show

configuration_InternLM_XComposer.py CHANGED Viewed

@@ -32,8 +32,8 @@ class InternLMXComposerConfig(PretrainedConfig):
         num_quant=32,
         intern_converted_llm=True,
         kqvo_bias=True,
-        # device='cuda:0',
-        device='cpu',
         internlm_lora=None,
         **kwargs,
     ):

         num_quant=32,
         intern_converted_llm=True,
         kqvo_bias=True,
+        device='cuda:0',
+        # device='cpu',
         internlm_lora=None,
         **kwargs,
     ):