Fixed an issue with samples being kept on the wrong device when no re…

…finer is used
SeargeDP · Aug 30, 2023 · fcd0fc3 · fcd0fc3
1 parent 40cf089
commit fcd0fc3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/custom_sdxl_ksampler.py b/modules/custom_sdxl_ksampler.py
@@ -194,7 +194,7 @@ def base_tonemap_reinhard(args):
     noise = torch.zeros(base_samples.size(), dtype=base_samples.dtype, layout=base_samples.layout, device=device)
 
     if refiner_steps < 1:
-        return base_samples
+        return base_samples.cpu()
 
     if refiner_detail_boost > 0.0:
         new_noise = comfy.sample.prepare_noise(original_latent, seed + 1, batch_inds).to(device)