Skip to content

Commit 85c1e42

Browse files
committed
fix: lora checkpoint
Signed-off-by: Mehant Kammakomati <[email protected]>
1 parent 3bf6a67 commit 85c1e42

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

plugins/accelerated-moe/src/fms_acceleration_moe/utils/checkpoint_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -677,7 +677,7 @@ def recover_safetensors_from_dcp(
677677
if name in lora_keys.keys():
678678
name = lora_keys[name]
679679
new_state_dict[name] = param
680-
680+
681681
# save it as a safetensors file
682682
save_sharded_safetensors(
683683
{k: v.contiguous() for k, v in new_state_dict.items()},

0 commit comments

Comments
 (0)