Fix error message in model_patcher.py
Found while tinkering.
This commit is contained in:
@@ -145,7 +145,7 @@ class ModelPatcher:
|
|||||||
model_sd = self.model_state_dict()
|
model_sd = self.model_state_dict()
|
||||||
for key in self.patches:
|
for key in self.patches:
|
||||||
if key not in model_sd:
|
if key not in model_sd:
|
||||||
print("could not patch. key doesn't exist in model:", k)
|
print("could not patch. key doesn't exist in model:", key)
|
||||||
continue
|
continue
|
||||||
|
|
||||||
weight = model_sd[key]
|
weight = model_sd[key]
|
||||||
|
|||||||
Reference in New Issue
Block a user