Merge c3cc84fa18 into 2307ff6746

* post method `/preview_method` is added

https://github.com/comfyanonymous/ComfyUI/issues/6205

comfy/sd.py

@@ -111,7 +111,7 @@ class CLIP:
             model_management.load_models_gpu([self.patcher], force_full_load=True)
         self.layer_idx = None
         self.use_clip_schedule = False
-        logging.info("CLIP model load device: {}, offload device: {}, current: {}, dtype: {}".format(load_device, offload_device, params['device'], dtype))
+        logging.info("CLIP/text encoder model load device: {}, offload device: {}, current: {}, dtype: {}".format(load_device, offload_device, params['device'], dtype))
 
     def clone(self):
         n = CLIP(no_init=True)
@@ -898,7 +898,7 @@ def load_state_dict_guess_config(sd, output_vae=True, output_clip=True, output_c
     if output_model:
         model_patcher = comfy.model_patcher.ModelPatcher(model, load_device=load_device, offload_device=model_management.unet_offload_device())
         if inital_load_device != torch.device("cpu"):
-            logging.info("loaded straight to GPU")
+            logging.info("loaded diffusion model directly to GPU")
             model_management.load_models_gpu([model_patcher], force_full_load=True)
 
     return (model_patcher, clip, vae, clipvision)

ruff.toml

@@ -4,7 +4,8 @@ lint.ignore = ["ALL"]
 # Enable specific rules
 lint.select = [
     "S307",  # suspicious-eval-usage
-    "T201",  # print-usage
+    "S102", # exec
+    "T",  # print-usage
     "W",
     # The "F" series in Ruff stands for "Pyflakes" rules, which catch various Python syntax errors and undefined names.
     # See all rules here: https://docs.astral.sh/ruff/rules/#pyflakes-f

server.py

@@ -33,6 +33,7 @@ from app.model_manager import ModelFileManager
 from app.custom_node_manager import CustomNodeManager
 from typing import Optional
 from api_server.routes.internal.internal_routes import InternalRoutes
+import latent_preview
 
 class BinaryEventTypes:
     PREVIEW_IMAGE = 1
@@ -692,6 +693,26 @@ class PromptServer():
 
             return web.Response(status=200)
 
+        @routes.post("/preview_method")
+        async def preview_method(request):
+            json_data =  await request.json()
+
+            if "value" in json_data:
+                method = json_data['value']
+
+                if method == 'auto':
+                    args.preview_method = latent_preview.LatentPreviewMethod.Auto
+                elif method == 'latent2rgb':
+                    args.preview_method = latent_preview.LatentPreviewMethod.Latent2RGB
+                elif method == 'taesd':
+                    args.preview_method = latent_preview.LatentPreviewMethod.TAESD
+                else:
+                    args.preview_method = latent_preview.LatentPreviewMethod.NoPreviews
+
+                return web.Response(status=200)
+
+            return web.Response(status=400)
+
     async def setup(self):
         timeout = aiohttp.ClientTimeout(total=None) # no timeout
         self.client_session = aiohttp.ClientSession(timeout=timeout)