stanrom commited on
Commit
6740366
1 Parent(s): f862375

Update modeling_InternLM_XComposer.py

Browse files
Files changed (1) hide show
  1. modeling_InternLM_XComposer.py +2 -2
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).to(torch.float16)
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,