Add cleanup only on the error exit (when tile < 128) before re-raising the OOM exception

comfy_extras/nodes_upscale_model.py

@@ -89,6 +89,7 @@ class ImageUpscaleWithModel(io.ComfyNode):
             except model_management.OOM_EXCEPTION as e:
                 tile //= 2
                 if tile < 128:
+                    upscale_model.to("cpu")
                     raise e
 
         # upscale_model.to("cpu")  # Commented out to keep model on GPU because when processing batch images then model unnecessarily moves to CPU