wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 14:50:49 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Fix model measurements to call

Browse Source
This commit is contained in:
doctorpangloss 2025-09-09 18:29:38 -07:00
parent 537e34358f
commit 67432bee6c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_management_types.py
View File

@ -287,9 +287,9 @@ class MemoryMeasurements:
@device.setter @device.setter
def device(self, value: torch.device): def device(self, value: torch.device):
if isinstance(self.model, DeviceSettable): if isinstance(self.model, DeviceSettable):
# todo: is this correct?
self.model.device = value self.model.device = value
elif hasattr(self.model, "to"): # todo: we don't want to `to` anything anymore here
self.model.to(value)
self._device = value self._device = value