From 524cd140b5373f1e4db8b3a5659adbafe11b4c9f Mon Sep 17 00:00:00 2001 From: patientx Date: Fri, 30 Aug 2024 13:33:32 +0300 Subject: [PATCH] removed bfloat from flux model support, resulting in 2x speedup --- 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 3603313fa..506205a4a 100644 --- a/comfy/supported_models.py +++ b/comfy/supported_models.py @@ -642,7 +642,7 @@ class Flux(supported_models_base.BASE): memory_usage_factor = 2.8 - supported_inference_dtypes = [torch.bfloat16, torch.float16, torch.float32] + supported_inference_dtypes = [torch.float16, torch.float32] vae_key_prefix = ["vae."] text_encoder_key_prefix = ["text_encoders."]