Update modeling_InternLM_XComposer.py
Browse files
modeling_InternLM_XComposer.py
CHANGED
@@ -66,9 +66,9 @@ class InternLMXComposerForCausalLM(PreTrainedModel):
|
|
66 |
assert int(torch.__version__[0]) == 2
|
67 |
# speed up init llm
|
68 |
with torch.device('meta'):
|
69 |
-
self.internlm_model = InternLMForCausalLM._from_config(config)
|
70 |
# self.internlm_model.to_empty(device=config.device).to(torch.float16)
|
71 |
-
self.internlm_model.tie_weights()
|
72 |
# self.internlm_model.to(config.device)
|
73 |
|
74 |
self.internlm_proj = nn.Linear(4096,
|
|
|
66 |
assert int(torch.__version__[0]) == 2
|
67 |
# speed up init llm
|
68 |
with torch.device('meta'):
|
69 |
+
self.internlm_model = InternLMForCausalLM._from_config(config)
|
70 |
# self.internlm_model.to_empty(device=config.device).to(torch.float16)
|
71 |
+
# self.internlm_model.tie_weights()
|
72 |
# self.internlm_model.to(config.device)
|
73 |
|
74 |
self.internlm_proj = nn.Linear(4096,
|