diff --git a/nodes_model_loading.py b/nodes_model_loading.py index f62c079..6ee9262 100644 --- a/nodes_model_loading.py +++ b/nodes_model_loading.py @@ -1345,7 +1345,6 @@ class WanVideoVAELoader: vae.to(device = offload_device, dtype = dtype) if compile_args is not None: vae.model.decoder = torch.compile(vae.model.decoder, fullgraph=compile_args["fullgraph"], dynamic=compile_args["dynamic"], backend=compile_args["backend"], mode=compile_args["mode"]) - print(vae) return (vae,)