Fix lint checks

pytorch · Dec 26, 2023 · 9c0df0a · 9c0df0a
1 parent 5488d72
commit 9c0df0a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/examples/large_models/gpt_fast/handler.py b/examples/large_models/gpt_fast/handler.py
@@ -50,10 +50,13 @@ def initialize(self, ctx):
         self.local_rank = rank if rank is not None else 0
 
         if torch.cuda.is_available():
-            if "torchrun" not in ctx.model_yaml_config and properties.get("gpu_id") is not None:
+            if (
+                "torchrun" not in ctx.model_yaml_config
+                and properties.get("gpu_id") is not None
+            ):
                 gpu_id = properties.get("gpu_id")
             else:
-                gpu_id = self.local_rank 
+                gpu_id = self.local_rank
             self.map_location = "cuda"
             self.device = torch.device(self.map_location + ":" + str(gpu_id))