cleanup

2026-03-13 00:31:56 +01:00
parent 4aed00afd9
commit 400e00418f
3 changed files with 3 additions and 3 deletions
@@ -1,3 +1,4 @@
+import contextlib
 import torch
 import functools

@@ -22,7 +23,6 @@ def safe_autocast_decorator(enabled=True):
        return wrapper
    return decorator

-import contextlib
@contextlib.contextmanager
 def safe_autocast(enabled=True):
    device = get_default_device()