diff --git a/comfy/model_management_types.py b/comfy/model_management_types.py index 2383f369a..259455bb3 100644 --- a/comfy/model_management_types.py +++ b/comfy/model_management_types.py @@ -287,9 +287,9 @@ class MemoryMeasurements: @device.setter def device(self, value: torch.device): if isinstance(self.model, DeviceSettable): + # todo: is this correct? self.model.device = value - elif hasattr(self.model, "to"): - self.model.to(value) + # todo: we don't want to `to` anything anymore here self._device = value