Update supported_models.py

This commit is contained in:
patientx 2024-12-23 02:54:49 +03:00 committed by GitHub
parent 713eca2176
commit 757335d901
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -757,7 +757,7 @@ class LTXV(supported_models_base.BASE):
memory_usage_factor = 2.7
supported_inference_dtypes = [torch.float16, torch.float32] #changed bfloat16 to float16 ZLUDA
supported_inference_dtypes = [torch.bfloat16, torch.float32]
vae_key_prefix = ["vae."]
text_encoder_key_prefix = ["text_encoders."]