tests: fix error with mixed_precision value on GH Actions

bghira · Sep 14, 2024 · 4dd81a2 · 4dd81a2
1 parent 558abad
commit 4dd81a2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_trainer.py b/tests/test_trainer.py
@@ -129,6 +129,7 @@ def test_stats_memory_used_none(
             timestep_bias_portion=0,
             metadata_update_interval=100,
             gradient_accumulation_steps=1,
+            mixed_precision="bf16",
             report_to="none",
             output_dir="output_dir",
         ),