Skip to content

Commit c36be0e

Browse files
Fix memory estimation bug with kontext. (#8709)
1 parent 9093301 commit c36be0e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/model_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -816,7 +816,7 @@ def extra_conds(self, **kwargs):
816816
class Flux(BaseModel):
817817
def __init__(self, model_config, model_type=ModelType.FLUX, device=None, unet_model=comfy.ldm.flux.model.Flux):
818818
super().__init__(model_config, model_type, device=device, unet_model=unet_model)
819-
self.memory_usage_factor_conds = ("kontext",)
819+
self.memory_usage_factor_conds = ("ref_latents",)
820820

821821
def concat_cond(self, **kwargs):
822822
try:

0 commit comments

Comments
 (0)