fix middle blocks temp conv

kabachuha · kabachuha · commit 5d5d66464f60 · 2023-04-17T19:28:48.000+03:00
diff --git a/utils/convert_diffusers_to_original_ms_text_to_video.py b/utils/convert_diffusers_to_original_ms_text_to_video.py
@@ -164,7 +164,7 @@
 unet_conversion_map_layer.append((sd_mid_atn_prefix, hf_mid_atn_prefix))
 
 for j in range(2):
-    hf_mid_res_prefix = f"mid_block.temp_convs.{j+1}."
+    hf_mid_res_prefix = f"mid_block.temp_convs.{j}."
     sd_mid_res_prefix = f"middle_block.{2*j+1}.temopral_conv."
     unet_conversion_map_layer.append((sd_mid_res_prefix, hf_mid_res_prefix))
 
