Update handler.py

handler.py

@@ -175,7 +175,9 @@ def initialize_vae_decoder(use_taehv=False, use_trt=False, device="cuda"):
         print("Initializing Default VAE Decoder...")
         vae_decoder = VAEDecoderWrapper()
         try:
-            vae_state_dict = torch.load('/repository/Wan2.1-T2V-1.3B/Wan2.1_VAE.pth', map_location="cpu")
+            # I should have called the folder "Wan2.1-T2V-1.3B" instead of "wan2.1"
+            #vae_state_dict = torch.load('/repository/Wan2.1-T2V-1.3B/Wan2.1_VAE.pth', map_location="cpu")
+            vae_state_dict = torch.load('/repository/wan2.1/Wan2.1_VAE.pth', map_location="cpu")
             decoder_state_dict = {k: v for k, v in vae_state_dict.items() if 'decoder.' in k or 'conv2' in k}
             vae_decoder.load_state_dict(decoder_state_dict)
         except FileNotFoundError: