From fb7280c3f4afefa73e0f2b1441a4af21667c9192 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Fri, 13 Jun 2025 13:40:44 +0300 Subject: [PATCH] Flux quanto fix logged dtype --- modules/model_flux.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/modules/model_flux.py b/modules/model_flux.py index 4381e1e50..3a5068377 100644 --- a/modules/model_flux.py +++ b/modules/model_flux.py @@ -35,11 +35,12 @@ def load_flux_quanto(checkpoint_info): quanto.requantize(transformer, state_dict, quantization_map, device=torch.device("cpu")) if shared.opts.diffusers_eval: transformer.eval() - if transformer.dtype != devices.dtype: + transformer_dtype = transformer.dtype + if transformer_dtype != devices.dtype: try: transformer = transformer.to(dtype=devices.dtype) except Exception: - shared.log.error(f"Load model: type=FLUX Failed to cast transformer to {devices.dtype}, set dtype to {transformer.dtype}") + shared.log.error(f"Load model: type=FLUX Failed to cast transformer to {devices.dtype}, set dtype to {transformer_dtype}") except Exception as e: shared.log.error(f"Load model: type=FLUX failed to load Quanto transformer: {e}") if debug: