Update model_list.py

This commit is contained in:
layerdiffusion
2024-08-14 16:49:31 -07:00
parent 3f96b28763
commit 84826248b4

View File

@@ -679,7 +679,7 @@ class Flux(BASE):
memory_usage_factor = 2.6
supported_inference_dtypes = [torch.bfloat16, torch.float32]
supported_inference_dtypes = [torch.bfloat16, torch.float16, torch.float32]
vae_key_prefix = ["vae."]
text_encoder_key_prefix = ["text_encoders."]