Fix comment formatting

wamartin-aml · Apr 14, 2021 · b947293 · b947293
1 parent 2a81d54
commit b947293
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/transformers/models/t5/modeling_t5.py b/src/transformers/models/t5/modeling_t5.py
@@ -645,7 +645,7 @@ def forward(
 
         # clamp inf values to enable fp16 training
         if self.ort:
-            #Remove data-based control flow for static graph
+            # Remove data-based control flow for static graph
             if hidden_states.dtype == torch.float16:
                 clamp_value = torch.where(torch.isinf(hidden_states).any(), torch.finfo(hidden_states.dtype).max - 1000,
                     torch.finfo(hidden_states.dtype).max)
@@ -679,7 +679,7 @@ def forward(
 
             # clamp inf values to enable fp16 training
             if self.ort:
-                #Remove data-based control flow for static graph
+                # Remove data-based control flow for static graph
                 if hidden_states.dtype == torch.float16:
                     clamp_value = torch.where(torch.isinf(hidden_states).any(), torch.finfo(hidden_states.dtype).max - 1000,
                         torch.finfo(hidden_states.dtype).max)
@@ -701,7 +701,7 @@ def forward(
 
         # clamp inf values to enable fp16 training
         if self.ort:
-            #Remove data-based control flow for static graph
+            # Remove data-based control flow for static graph
             if hidden_states.dtype == torch.float16:
                 clamp_value = torch.where(torch.isinf(hidden_states).any(), torch.finfo(hidden_states.dtype).max - 1000,
                     torch.finfo(hidden_states.dtype).max)