Update latent_preview.py

This commit is contained in:
taabata
2024-11-18 03:40:52 +03:00
committed by GitHub
parent d6e61db4a4
commit 0a161088dd
+2 -3
View File
@@ -10,7 +10,7 @@ import comfy.utils
import logging import logging
from pathlib import Path from pathlib import Path
from urllib import request from urllib import request
import base64, io, json import base64, io, json, os
MAX_PREVIEW_RESOLUTION = args.preview_size MAX_PREVIEW_RESOLUTION = args.preview_size
@@ -33,7 +33,7 @@ def preview_to_image(latent_image,newsample=False):
} }
} }
data = json.dumps(p).encode('utf-8') data = json.dumps(p).encode('utf-8')
req = request.Request("http://localhost:5000/settaesd", data=data) req = request.Request(f"http://{args.listen}:{os.environ['CANVASPORT']}/settaesd", data=data)
req.add_header("Content-Type", "application/json") req.add_header("Content-Type", "application/json")
request.urlopen(req) request.urlopen(req)
except Exception as e: except Exception as e:
@@ -125,4 +125,3 @@ def prepare_callback(model, steps, x0_output_dict=None):
preview_bytes = previewer.decode_latent_to_preview_image(preview_format, x0) preview_bytes = previewer.decode_latent_to_preview_image(preview_format, x0)
pbar.update_absolute(step + 1, total_steps, preview_bytes) pbar.update_absolute(step + 1, total_steps, preview_bytes)
return callback return callback