Skip to content

Commit b6ae0f2

Browse files
committed
fixing conversion script
1 parent fb20038 commit b6ae0f2

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/maxdiffusion/scripts/convert_ltx2_vae_weights.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ def convert_ltx2_vae(hf_repo, output_path):
8686
continue
8787

8888
pt_tensor = pt_state_dict[pt_key]
89-
np_array = pt_tensor.numpy()
89+
np_array = pt_tensor.float().numpy()
9090

9191
# Handle shape mismatch (Transpose Conv3d weights)
9292
is_conv_weight = "conv" in pt_key and "weight" in pt_key and len(np_array.shape) == 5

0 commit comments

Comments
 (0)