change cuda
Browse files
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 |
# 检查形状是否匹配,以避免错误
|