fix

ydshieh · ydshieh · commit cb42793402fc · 2023-06-16T16:04:21.000+02:00
diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py
@@ -2595,6 +2595,8 @@ def autocast_smart_context_manager(self, cache_enabled: Optional[bool] = True):
                 if self.use_cpu_amp
                 else torch.cuda.amp.autocast(cache_enabled=cache_enabled, dtype=self.amp_dtype)
             )
+        else:
+            ctx_manager = contextlib.nullcontext() if sys.version_info >= (3, 7) else contextlib.suppress()
 
         return ctx_manager
 

Original file line number	Diff line number	Diff line change
`@@ -2595,6 +2595,8 @@ def autocast_smart_context_manager(self, cache_enabled: Optional[bool] = True):`
`2595`	`2595`	`if self.use_cpu_amp`
`2596`	`2596`	`else torch.cuda.amp.autocast(cache_enabled=cache_enabled, dtype=self.amp_dtype)`
`2597`	`2597`	`)`
	`2598`	`+ else:`
	`2599`	`+ ctx_manager = contextlib.nullcontext() if sys.version_info >= (3, 7) else contextlib.suppress()`
`2598`	`2600`
`2599`	`2601`	`return ctx_manager`
`2600`	`2602`