Change default serialization for push_to_hf_hub to 'both'

huggingface · Jun 7, 2024 · 4535a54 · 4535a54
1 parent 5cce218
commit 4535a54
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/timm/models/_hub.py b/timm/models/_hub.py
@@ -285,7 +285,7 @@ def push_to_hf_hub(
         model_config: Optional[dict] = None,
         model_card: Optional[dict] = None,
         model_args: Optional[dict] = None,
-        safe_serialization: Union[bool, Literal["both"]] = False,
+        safe_serialization: Union[bool, Literal["both"]] = 'both',
 ):
     """
     Arguments: