keras-sd
/

text-encoder

chansung commited on Dec 16, 2022

Commit

a57f986

1 Parent(s): f88c9df

add custom handler

Files changed (2) hide show

__pycache__/handler.cpython-38.pyc CHANGED Viewed

Binary files a/__pycache__/handler.cpython-38.pyc and b/__pycache__/handler.cpython-38.pyc differ

handler.py CHANGED Viewed

@@ -73,4 +73,4 @@ class EndpointHandler():
         unconditional_context_b64 = base64.b64encode(unconditional_context.numpy().tobytes())
         unconditional_context_b64str = unconditional_context_b64.decode()
-        return context_b64str, unconditional_context_b64str

         unconditional_context_b64 = base64.b64encode(unconditional_context.numpy().tobytes())
         unconditional_context_b64str = unconditional_context_b64.decode()
+        return {"context_b64str": context_b64str, "unconditional_context_b64str": unconditional_context_b64str}