[diffusion]: add dummy device attribute to fix AttributeError (#17949)

This commit is contained in:
Ratish P
2026-01-29 23:05:12 +05:30
committed by GitHub
parent 0769de9b0f
commit 88fb927cc9

View File

@@ -79,6 +79,7 @@ class Platform:
_enum: PlatformEnum
device_name: str
device_type: str
device: torch.device | None = None # Dummy attribute for compatibility
# available dispatch keys:
# check https://github.com/pytorch/pytorch/blob/313dac6c1ca0fa0cde32477509cce32089f8532a/torchgen/model.py#L134 # noqa