zixiao commited on
Commit ·
f924f8a
1
Parent(s): d774b46
update.
Browse files- convert.py +1 -1
convert.py
CHANGED
|
@@ -24,7 +24,7 @@ for k, v in state_dict.items():
|
|
| 24 |
#print(old_prefix, new_prefix)
|
| 25 |
if k.startswith(old_prefix):
|
| 26 |
key = new_prefix + k[len(old_prefix):]
|
| 27 |
-
print(f"{k} -> {key}")
|
| 28 |
break
|
| 29 |
else:
|
| 30 |
key = k
|
|
|
|
| 24 |
#print(old_prefix, new_prefix)
|
| 25 |
if k.startswith(old_prefix):
|
| 26 |
key = new_prefix + k[len(old_prefix):]
|
| 27 |
+
print(f"{k} -> {key} sum:{v.sum()}")
|
| 28 |
break
|
| 29 |
else:
|
| 30 |
key = k
|