Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-11-04 14:16:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

added better missing model error handling

Browse Source
This commit is contained in:
deepbeepmeep 2025-07-16 00:49:16 +02:00
parent bda410f367
commit 1e2d74ae7d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
wgp.py
View File

@ -2132,8 +2132,8 @@ for model_type in model_types:
transformer_types = server_config.get("transformer_types", []) transformer_types = server_config.get("transformer_types", [])
new_transformer_types = [] new_transformer_types = []
for model_type in transformer_types: for model_type in transformer_types:
if get_model_def(model_type) == None: if get_model_def(model_type) == None:
print(f"Model '{model_type}' is missing. Either install it in the finetune folder or remove this model in wgp_config.json") print(f"Model '{model_type}' is missing. Either install it in the finetune folder or remove this model from ley 'transformer_types' in wgp_config.json")
else: else:
new_transformer_types.append(model_type) new_transformer_types.append(model_type)
transformer_types = new_transformer_types transformer_types = new_transformer_types