We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3bf6a67 commit 85c1e42Copy full SHA for 85c1e42
1 file changed
plugins/accelerated-moe/src/fms_acceleration_moe/utils/checkpoint_utils.py
@@ -677,7 +677,7 @@ def recover_safetensors_from_dcp(
677
if name in lora_keys.keys():
678
name = lora_keys[name]
679
new_state_dict[name] = param
680
-
+
681
# save it as a safetensors file
682
save_sharded_safetensors(
683
{k: v.contiguous() for k, v in new_state_dict.items()},
0 commit comments