feat: add check for existence of accelerator before waiting

EvolvingLMMs-Lab · Sep 5, 2024 · e1df331 · e1df331
1 parent 5b310f4
commit e1df331
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lmms_eval/evaluator.py b/lmms_eval/evaluator.py
@@ -607,7 +607,8 @@ def evaluate(
     else:
         results_dict = None
 
-    lm.accelerator.wait_for_everyone()
+    if hasattr(lm, "accelerator"):
+        lm.accelerator.wait_for_everyone()
     return results_dict