Fixes a typo default arg (#604)

mosaicml · Sep 18, 2023 · 1e7f909 · 1e7f909
1 parent 229ab4f
commit 1e7f909
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
@@ -46,7 +46,7 @@ def __init__(
         save_folder: str,
         save_interval: Union[str, int, Time],
         huggingface_folder_name: str = 'ba{batch}',
-        precision: str = 'fp32',
+        precision: str = 'float32',
         overwrite: bool = False,
     ):
         self.backend, self.bucket_name, self.save_dir_format_str = parse_uri(

diff --git a/tests/test_hf_conversion_script.py b/tests/test_hf_conversion_script.py
@@ -176,6 +176,10 @@ def get_config(
     return cast(DictConfig, test_cfg)
 
 
+def test_callback_inits_with_defaults():
+    _ = HuggingFaceCheckpointer(save_folder='test', save_interval='1ba')
+
+
 @pytest.mark.world_size(2)
 @pytest.mark.gpu
 @pytest.mark.parametrize('model', ['mpt', 'neo', 'llama2'])