Update nodes_model_loading.py
This commit is contained in:
@@ -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,)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user