Upodate model tag on loras

This commit is contained in:
Jaret Burkett
2025-04-19 10:41:27 -06:00
parent d455e76c4f
commit 77001ee77f
6 changed files with 35 additions and 12 deletions

View File

@@ -1486,3 +1486,7 @@ class BaseModel:
def get_transformer_block_names(self) -> Optional[List[str]]:
# override in child classes to get transformer block names for lora targeting
return None
def get_base_model_version() -> str:
# override in child classes to get the base model version
return "unknown"

View File

@@ -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"