Update latent_preview.py
This commit is contained in:
+2
-3
@@ -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
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user