Skip to content
This repository was archived by the owner on Dec 14, 2023. It is now read-only.

Commit d65cf47

Browse files
committed
finally fix all the keys carryover
1 parent 3a418d8 commit d65cf47

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

utils/convert_diffusers_to_original_ms_text_to_video.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@
142142

143143
# no upsample in up_blocks.3
144144
hf_upsample_prefix = f"up_blocks.{i}.upsamplers.0."
145-
sd_upsample_prefix = f"output_blocks.{3*i + 2}.{1 if i == 0 else 2}."
145+
sd_upsample_prefix = f"output_blocks.{3*i + 2}.{1 if i == 0 else 3}."
146146
unet_conversion_map_layer.append((sd_upsample_prefix, hf_upsample_prefix))
147147

148148

@@ -155,7 +155,7 @@
155155

156156
for j in range(2):
157157
hf_mid_res_prefix = f"mid_block.resnets.{j}."
158-
sd_mid_res_prefix = f"middle_block.{2*j+1}."
158+
sd_mid_res_prefix = f"middle_block.{3*j}."
159159
unet_conversion_map_layer.append((sd_mid_res_prefix, hf_mid_res_prefix))
160160

161161
# Temporal

0 commit comments

Comments
 (0)