298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00: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 5cbb01bc2f
commit 9ee0a6553a
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 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."] vae_key_prefix = ["vae."]
text_encoder_key_prefix = ["text_encoders."] text_encoder_key_prefix = ["text_encoders."]