From 9512cac9d2e98e40be75244d7130e0ec0d6a2c7e Mon Sep 17 00:00:00 2001 From: root Date: Fri, 28 Jul 2023 05:59:11 +0000 Subject: [PATCH] Added jsonify base64 --- build_docker.sh | 5 +++++ comfy_runpod.py | 5 +++-- 2 files changed, 8 insertions(+), 2 deletions(-) create mode 100755 build_docker.sh diff --git a/build_docker.sh b/build_docker.sh new file mode 100755 index 000000000..1101c284f --- /dev/null +++ b/build_docker.sh @@ -0,0 +1,5 @@ +docker build -t comfy . + +docker tag comfy forme2/comfy + +docker push forme2/comfy diff --git a/comfy_runpod.py b/comfy_runpod.py index 967b362c8..6a29b4c88 100644 --- a/comfy_runpod.py +++ b/comfy_runpod.py @@ -74,10 +74,11 @@ def run_prompt(job): for node_id in images: for image_data in images[node_id]: image = Image.open(io.BytesIO(image_data)) - im_file = BytesIO() + im_file = io.BytesIO() image.save(im_file, format="JPEG") im_bytes = im_file.getvalue() # im_bytes: image in binary format. - im_b64 = base64.b64encode(im_bytes) + im_b64 = base64.b64encode(im_bytes) + im_b64 = str(im_b64) data['images'].append(im_b64) return data