the torch compile for vae will make the process failed. This exception is raised TypeError: decode() missing 1 required positional argument: 'num_frames' After torch compile, the vae forward function no longer have num_frames arg, instead it becomes args and kwargs. I just comment out that line for now.