Skip to content

Commit 6af308f

Browse files
committed
add import for `_TORCH_GREATER_EQUAL_2_3
1 parent 835c386 commit 6af308f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/lightning/fabric/utilities/device_dtype_mixin.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@
1818
from torch.nn import Module
1919
from typing_extensions import Self, override
2020

21+
from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_2_3
22+
2123

2224
class _DeviceDtypeModuleMixin(Module):
2325
__jit_unused_properties__: list[str] = ["device", "dtype"]

0 commit comments

Comments
 (0)