From 67432bee6c63d24f04ea8a4d3cb1e71fe48b6860 Mon Sep 17 00:00:00 2001 From: doctorpangloss <@hiddenswitch.com> Date: Tue, 9 Sep 2025 18:29:38 -0700 Subject: [PATCH] Fix model measurements `to` call --- comfy/model_management_types.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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