From e5f70c4d78af6f690fe556aa4531c4e9d196b160 Mon Sep 17 00:00:00 2001 From: xmarre Date: Tue, 31 Mar 2026 19:26:26 +0200 Subject: [PATCH] Restore default forecast device refresh behavior --- comfyui_spectrum/forecast.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfyui_spectrum/forecast.py b/comfyui_spectrum/forecast.py index 17bc315..210f269 100644 --- a/comfyui_spectrum/forecast.py +++ b/comfyui_spectrum/forecast.py @@ -91,11 +91,11 @@ class ChebyshevSpectrumForecaster: self._feature_dtype = output_dtype if output_dtype is not None else feat.dtype if predict_device is not None: self._predict_device = predict_device - elif self._predict_device is None: + else: self._predict_device = feat.device if output_device is not None: self._output_device = output_device - elif self._output_device is None: + else: self._output_device = feat.device if self._predict_device is None: self._predict_device = self._output_device