jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Fix memory estimation bug with kontext. (#8709)

Browse Source
This commit is contained in:
comfyanonymous
2025-06-27 14:21:12 -07:00
committed by GitHub
parent 9093301a49
commit c36be0ea09
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_base.py
View File

@@ -816,7 +816,7 @@ class PixArt(BaseModel):
class Flux(BaseModel): class Flux(BaseModel):
def __init__(self, model_config, model_type=ModelType.FLUX, device=None, unet_model=comfy.ldm.flux.model.Flux): def __init__(self, model_config, model_type=ModelType.FLUX, device=None, unet_model=comfy.ldm.flux.model.Flux):
super().__init__(model_config, model_type, device=device, unet_model=unet_model) super().__init__(model_config, model_type, device=device, unet_model=unet_model)
self.memory_usage_factor_conds = ("kontext",) self.memory_usage_factor_conds = ("ref_latents",)
def concat_cond(self, **kwargs): def concat_cond(self, **kwargs):
try: try: