⚗️ | Small fixes here and there

2025-12-11 23:06:38 +02:00
parent e3e555794e
commit 571c403b93
4 changed files with 12 additions and 13 deletions
--- a/training.py
+++ b/training.py
@@ -34,7 +34,11 @@ args = parser.parse_args()
 # Init accelerator
 # ---------------------------

-accelerator = Accelerator(mixed_precision="bf16")
+try:
+    accelerator = Accelerator(mixed_precision="bf16")
+except Exception:
+    accelerator = Accelerator(mixed_precision="fp16")
+    accelerator.print("⚠️ | bf16 unavailable — falling back to fp16")

 # ---------------------------
 # Models