Update configuration_InternLM_XComposer.py
Browse files
configuration_InternLM_XComposer.py
CHANGED
|
@@ -32,8 +32,8 @@ class InternLMXComposerConfig(PretrainedConfig):
|
|
| 32 |
num_quant=32,
|
| 33 |
intern_converted_llm=True,
|
| 34 |
kqvo_bias=True,
|
| 35 |
-
|
| 36 |
-
device='cpu',
|
| 37 |
internlm_lora=None,
|
| 38 |
**kwargs,
|
| 39 |
):
|
|
|
|
| 32 |
num_quant=32,
|
| 33 |
intern_converted_llm=True,
|
| 34 |
kqvo_bias=True,
|
| 35 |
+
device='cuda:0',
|
| 36 |
+
# device='cpu',
|
| 37 |
internlm_lora=None,
|
| 38 |
**kwargs,
|
| 39 |
):
|