wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-01 17:20:17 +08:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

Update supported inference dtypes in supported_models.py

Browse Source
This commit is contained in:
patientx 2025-12-22 02:19:49 +03:00 committed by GitHub
parent 2edfcd6141
commit 341a19e500
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/supported_models.py
View File

@ -428,7 +428,7 @@ class Stable_Cascade_C(supported_models_base.BASE):
unet_extra_config = {}
latent_format = latent_formats.SC_Prior
supported_inference_dtypes = [torch.float16, torch.float32]
supported_inference_dtypes = [torch.bfloat16, torch.float32]
sampling_settings = {
"shift": 2.0,
@ -474,7 +474,7 @@ class Stable_Cascade_B(Stable_Cascade_C):
unet_extra_config = {}
latent_format = latent_formats.SC_B
supported_inference_dtypes = [torch.float16, torch.float32]
supported_inference_dtypes = [torch.float16, torch.bfloat16, torch.float32]
sampling_settings = {
"shift": 1.0,
@ -706,7 +706,7 @@ class Flux(supported_models_base.BASE):
memory_usage_factor = 3.1 # TODO: debug why flux mem usage is so weird on windows.
supported_inference_dtypes = [torch.float16, torch.float32]
supported_inference_dtypes = [torch.bfloat16, torch.float16, torch.float32]
vae_key_prefix = ["vae."]
text_encoder_key_prefix = ["text_encoders."]