From 9ee0a6553ab94f221789d1905af66435bb1b2b96 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Sun, 27 Oct 2024 04:56:40 -0400 Subject: [PATCH] float16 inference is a bit broken on mochi. --- comfy/supported_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/supported_models.py b/comfy/supported_models.py index 57099082..8300c4c3 100644 --- a/comfy/supported_models.py +++ b/comfy/supported_models.py @@ -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."]