Support mixed precision fp8 scaled models
This commit is contained in:
@@ -1480,6 +1480,8 @@ class WanVideoModelLoader:
|
|||||||
sd.update(extra_sd)
|
sd.update(extra_sd)
|
||||||
del extra_sd
|
del extra_sd
|
||||||
|
|
||||||
|
sd = {k.replace(".weight_scale", ".scale_weight"): v for k, v in sd.items()}
|
||||||
|
|
||||||
# FlashVSR
|
# FlashVSR
|
||||||
if "LQ_proj_in.norm1.gamma" in sd:
|
if "LQ_proj_in.norm1.gamma" in sd:
|
||||||
log.info("FlashVSR model detected, patching model...")
|
log.info("FlashVSR model detected, patching model...")
|
||||||
|
|||||||
Reference in New Issue
Block a user