mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-07-09 08:47:17 +08:00
Fix memory estimation bug with kontext. (#8709)
This commit is contained in:
parent
9093301a49
commit
c36be0ea09
@ -816,7 +816,7 @@ class PixArt(BaseModel):
|
||||
class Flux(BaseModel):
|
||||
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)
|
||||
self.memory_usage_factor_conds = ("kontext",)
|
||||
self.memory_usage_factor_conds = ("ref_latents",)
|
||||
|
||||
def concat_cond(self, **kwargs):
|
||||
try:
|
||||
|
Loading…
x
Reference in New Issue
Block a user