Upodate model tag on loras

2025-04-19 10:41:27 -06:00
parent d455e76c4f
commit 77001ee77f
6 changed files with 35 additions and 12 deletions
--- a/toolkit/models/wan21/wan21.py
+++ b/toolkit/models/wan21/wan21.py
@@ -663,3 +663,6 @@ class Wan21(BaseModel):

    def convert_lora_weights_before_load(self, state_dict):
        return convert_to_diffusers(state_dict)
+    
+    def get_base_model_version(self):
+        return "wan_2.1"