[diffusion]: add dummy device attribute to fix AttributeError (#17949)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user