Clean

huggingface · Nov 10, 2023 · 2ca0973 · 2ca0973
1 parent aff488b
commit 2ca0973
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/accelerate/utils/other.py b/src/accelerate/utils/other.py
@@ -152,6 +152,7 @@ def clean_state_dict_for_safetensors(state_dict: dict):
     state_dict = {k: v.contiguous() for k, v in state_dict.items()}
     return state_dict
 
+
 def save(obj, f, save_on_each_node: bool = False, safe_serialization: bool = False):
     """
     Save the data to disk. Use in place of `torch.save()`.