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

fixed bug while loading fp16 model

Browse Source
This commit is contained in:
deepbeepmeep 2025-06-14 10:05:11 +02:00 committed by GitHub
parent d5f9170454
commit 6c8779def6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
wgp.py
View File

@ -1746,7 +1746,7 @@ def get_model_filename(model_type, quantization ="int8", dtype_policy = ""):
else: else:
raw_filename = choices[0] raw_filename = choices[0]
if dtype == torch.float16 and not any("fp16","FP16") in raw_filename and model_family == "wan" and finetune_def == None : if dtype == torch.float16 and not any(x in raw_filename for x in ["fp16", "FP16"]) and model_family == "wan" and finetune_def == None:
if "quanto_int8" in raw_filename: if "quanto_int8" in raw_filename:
raw_filename = raw_filename.replace("quanto_int8", "quanto_fp16_int8") raw_filename = raw_filename.replace("quanto_int8", "quanto_fp16_int8")
elif "quanto_bf16_int8" in raw_filename: elif "quanto_bf16_int8" in raw_filename: