mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-06-03 21:07:27 +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
|
return
|
||||||
try:
|
try:
|
||||||
if not state["loaded"]:
|
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
|
state["loaded"] = True
|
||||||
img = decode_x0_to_image(vae, x0, cfg)
|
img = decode_x0_to_image(vae, x0, cfg)
|
||||||
if state["pbar"] is None:
|
if state["pbar"] is None:
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user