Update supported_models.py

This commit is contained in:
patientx 2025-02-05 23:12:38 +03:00 committed by GitHub
parent f8c2ab631a
commit 93c0fc3446
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -420,7 +420,7 @@ class Stable_Cascade_C(supported_models_base.BASE):
unet_extra_config = {}
latent_format = latent_formats.SC_Prior
supported_inference_dtypes = [torch.bfloat16, torch.float32]
supported_inference_dtypes = [torch.float16, torch.float32]
sampling_settings = {
"shift": 2.0,
@ -466,7 +466,7 @@ class Stable_Cascade_B(Stable_Cascade_C):
unet_extra_config = {}
latent_format = latent_formats.SC_B
supported_inference_dtypes = [torch.float16, torch.bfloat16, torch.float32]
supported_inference_dtypes = [torch.float16, torch.float32]
sampling_settings = {
"shift": 1.0,