Update configuration_internlm_xcomposer2.py
Browse files
configuration_internlm_xcomposer2.py
CHANGED
|
@@ -108,7 +108,7 @@ class InternLMXcomposer2Config(PretrainedConfig):
|
|
| 108 |
rope_theta=10000,
|
| 109 |
rope_scaling=None,
|
| 110 |
attn_implementation='eager',
|
| 111 |
-
device='cuda',
|
| 112 |
# dtype='bfloat16',
|
| 113 |
**kwargs,
|
| 114 |
):
|
|
|
|
| 108 |
rope_theta=10000,
|
| 109 |
rope_scaling=None,
|
| 110 |
attn_implementation='eager',
|
| 111 |
+
device='cuda:0',
|
| 112 |
# dtype='bfloat16',
|
| 113 |
**kwargs,
|
| 114 |
):
|