xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-07 15:57:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Adjust memory usage factor and remove unnecessary code

Browse Source
This commit is contained in:
silveroxides 2025-04-08 12:48:25 +02:00
parent c4f6874dc1
commit e1af413722
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/supported_models.py
View File

@ -1034,18 +1034,16 @@ class Chroma(supported_models_base.BASE):
} }
sampling_settings = { sampling_settings = {
"multiplier": 1.0,
"shift": 1.0,
} }
latent_format = comfy.latent_formats.Flux latent_format = comfy.latent_formats.Flux
memory_usage_factor = 1.8 memory_usage_factor = 3.2
supported_inference_dtypes = [torch.bfloat16, torch.float16, torch.float32] supported_inference_dtypes = [torch.bfloat16, torch.float16, torch.float32]
def get_model(self, state_dict, prefix="", device=None): def get_model(self, state_dict, prefix="", device=None):
out = model_base.Chroma(self, model_type=model_base.ModelType.FLOW, device=device) out = model_base.Chroma(self, device=device)
return out return out
def clip_target(self, state_dict={}): def clip_target(self, state_dict={}):