fix bug in run

EleutherAI · Aug 31, 2023 · 8126b4a · 8126b4a
1 parent 4e00e48
commit 8126b4a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/elk/run.py b/elk/run.py
@@ -97,6 +97,8 @@ def execute(
             )
 
         devices = select_usable_devices(self.num_gpus, min_memory=self.min_gpu_mem)
+        if devices == ['mps']:
+            devices = ['cpu']
         num_devices = len(devices)
         func: Callable[[int], dict[str, pd.DataFrame]] = partial(
             self.apply_to_layer, devices=devices, world_size=num_devices