fix test suite

bghira · Oct 1, 2024 · eafb09e · eafb09e
1 parent fec2827
commit eafb09e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/helpers/training/default_settings/safety_check.py b/helpers/training/default_settings/safety_check.py
@@ -26,7 +26,7 @@ def safety_check(args, accelerator):
 
     if (
         (args.base_model_precision in ["fp8-quanto", "int4-quanto"] or (args.base_model_precision != "no_change" and args.quantize_activations))
-        and accelerator.state.dynamo_plugin.backend.lower() == "inductor"
+        and (accelerator is not None and accelerator.state.dynamo_plugin.backend.lower() == "inductor")
     ):
         logger.warning(
             f"{args.base_model_precision} is not supported with Dynamo backend. Disabling Dynamo."