fix flake

google · Jul 6, 2024 · 33218cf · 33218cf
1 parent e3f63e9
commit 33218cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/DiarizationLM/unsloth/2_export.py b/DiarizationLM/unsloth/2_export.py
@@ -6,8 +6,8 @@
 
 
 def export_models(
-    save_lora: bool=True,
-    save_16bit: bool=True,
+    save_lora: bool = True,
+    save_16bit: bool = True,
 ) -> None:
   ############################################################################
   # Get model