298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-17 09:33:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add supported_inference_dtypes

Browse Source
This commit is contained in:
Silver 2025-03-22 16:49:16 +01:00 committed by GitHub
parent ca73500269
commit fda35f37e5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/supported_models.py
View File

@ -1027,6 +1027,7 @@ class Chroma(supported_models_base.BASE):
}
latent_format = comfy.latent_formats.Flux
memory_usage_factor = 2.8
supported_inference_dtypes = [torch.bfloat16, torch.float16, torch.float32]
def get_model(self, state_dict, prefix="", device=None):
out = model_base.Chroma(self, model_type=model_base.ModelType.FLUX, device=device)