Upodate model tag on loras
This commit is contained in:
@@ -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"
|
||||
|
||||
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user