jcccy commited on
Commit
1a2dde2
·
1 Parent(s): dee39c5

change cuda

Browse files
Files changed (1) hide show
  1. minigpt4/models/minigpt4.py +1 -0
minigpt4/models/minigpt4.py CHANGED
@@ -451,6 +451,7 @@ class MiniGPT4(MiniGPTBase):
451
  state_dict_a = ckpt['model']
452
  state_dict_b = model.state_dict()
453
  for name_b, param_b in state_dict_b.items():
 
454
  if name_b in state_dict_a:
455
  param_a = state_dict_a[name_b]
456
  # 检查形状是否匹配,以避免错误
 
451
  state_dict_a = ckpt['model']
452
  state_dict_b = model.state_dict()
453
  for name_b, param_b in state_dict_b.items():
454
+ print(name_b,param_b.shape)
455
  if name_b in state_dict_a:
456
  param_a = state_dict_a[name_b]
457
  # 检查形状是否匹配,以避免错误