mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-06-02 20:37:35 +08:00
Fix issue with triposplat preview and old offloading mode. (#14218)
This commit is contained in:
parent
e785f0d212
commit
06b710aa68
@ -233,7 +233,9 @@ class TripoSplatSamplingPreview(IO.ComfyNode):
|
||||
return
|
||||
try:
|
||||
if not state["loaded"]:
|
||||
comfy.model_management.load_models_gpu([vae.patcher], memory_required=memory_required)
|
||||
loaded_models = comfy.model_management.loaded_models(only_currently_used=True)
|
||||
loaded_models.append(vae.patcher)
|
||||
comfy.model_management.load_models_gpu(loaded_models, memory_required=memory_required)
|
||||
state["loaded"] = True
|
||||
img = decode_x0_to_image(vae, x0, cfg)
|
||||
if state["pbar"] is None:
|
||||
|
||||
Loading…
Reference in New Issue
Block a user