xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-19 09:36:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

float16 inference is a bit broken on mochi.

Browse Source
This commit is contained in:
comfyanonymous 2024-10-27 04:56:40 -04:00
parent 9f8e7912ef
commit 1a27859ed1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/supported_models.py
View File

@ -686,7 +686,7 @@ class GenmoMochi(supported_models_base.BASE):
memory_usage_factor = 2.0 #TODO
supported_inference_dtypes = [torch.float16, torch.bfloat16, torch.float32]
supported_inference_dtypes = [torch.bfloat16, torch.float32]
vae_key_prefix = ["vae."]
text_encoder_key_prefix = ["text_encoders."]