more middle block weights carryover

kabachuha · kabachuha · commit 3a418d810f17 · 2023-04-17T19:44:34.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
@@ -165,7 +165,7 @@
 
 for j in range(2):
     hf_mid_res_prefix = f"mid_block.temp_convs.{j}."
-    sd_mid_res_prefix = f"middle_block.{2*j+1}.temopral_conv."
+    sd_mid_res_prefix = f"middle_block.{3*j}.temopral_conv."
     unet_conversion_map_layer.append((sd_mid_res_prefix, hf_mid_res_prefix))
 
 # The pipeline
