Update save.py

2024-12-26 01:44:12 -08:00 · 2024-12-26 01:44:12 -08:00 · c8d2f5b8da
commit c8d2f5b8da
parent ae636f21c9
1 changed files with 1 additions and 0 deletions
--- a/unsloth/save.py
+++ b/unsloth/save.py
@ -2136,6 +2136,7 @@ def unsloth_generic_save(
        push_to_hub          = push_to_hub,
        private              = private,
        token                = token,
+        output_dtype         = None,
        low_disk_space_usage = False,
        use_temp_file        = False,
    )