We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 835c386 commit 6af308fCopy full SHA for 6af308f
src/lightning/fabric/utilities/device_dtype_mixin.py
@@ -18,6 +18,8 @@
18
from torch.nn import Module
19
from typing_extensions import Self, override
20
21
+from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_2_3
22
+
23
24
class _DeviceDtypeModuleMixin(Module):
25
__jit_unused_properties__: list[str] = ["device", "dtype"]
0 commit comments