fix the accelerator issue

Co-authored-by: eigen2017 <[email protected]>
microsoft · Apr 15, 2024 · 2aca1aa · 2aca1aa
1 parent c89600c
commit 2aca1aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/accelerator/mps_accelerator.py b/accelerator/mps_accelerator.py
@@ -260,7 +260,7 @@ def export_envs(self):
         return []
 
     def visible_devices_envs(self):
-        # TODO could not find visible devices env for mps
+        # TODO: could not find visible devices env for mps
         return ['CUDA_VISIBLE_DEVICES']
 
     def set_visible_devices_envs(self, current_env, local_accelerator_ids):