Convert wan lora weights on save to be something comfy can handle
This commit is contained in:
1 parent
7e37918fbc
commit
e6739f7eb2
8 files changed
+108
-5
No files matched your search
@@ -3061,3 +3061,11 @@ class StableDiffusion:
|
||||
encoder.to(*args, **kwargs)
|
||||
else:
|
||||
self.text_encoder.to(*args, **kwargs)
|
||||
|
||||
def convert_lora_weights_before_save(self, state_dict):
|
||||
# can be overridden in child classes to convert weights before saving
|
||||
return state_dict
|
||||
|
||||
def convert_lora_weights_before_load(self, state_dict):
|
||||
# can be overridden in child classes to convert weights before loading
|
||||
return state_dict
|
||||
Reference in new issue
Block a user