nodes/modelscope-scepter
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

v1.0.3 update

Browse Source
This commit is contained in:
zeyinzi.jzyz
2024-07-18 14:12:42 +08:00
parent 7a9f90efb2
commit 01fd8335af
94 changed files with 8776 additions and 417 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scepter/modules/utils/model.py
+4 -4
View File
@@ -134,11 +134,11 @@ def load_pretrained_dict(model: torch.nn.Module,
missing_keys = load_status.missing_keys
err_msgs = []
if unexpected_keys:
err_msgs.append('unexpected key in source '
f'state_dict: {", ".join(unexpected_keys)}\n')
err_msgs.append('unexpected key in source state_dict: {}\n'.format(
', '.join(unexpected_keys)))
if missing_keys:
err_msgs.append('missing key in source '
f'state_dict: {", ".join(missing_keys)}\n')
err_msgs.append('missing key in source state_dict: {}\n'.format(
', '.join(missing_keys)))
err_msgs = '\n'.join(err_msgs)
if len(err_msgs) > 0: