Update handler.py
Browse files- handler.py +3 -1
handler.py
CHANGED
@@ -62,10 +62,12 @@ class EndpointHandler():
|
|
62 |
context = base64.b64decode(contexts[0])
|
63 |
context = np.frombuffer(context, dtype="float32")
|
64 |
context = np.reshape(context, (batch_size, 77, 768))
|
|
|
65 |
|
66 |
unconditional_context = base64.b64decode(contexts[1])
|
67 |
unconditional_context = np.frombuffer(unconditional_context, dtype="float32")
|
68 |
-
unconditional_context = np.reshape(unconditional_context, (batch_size, 77, 768))
|
|
|
69 |
|
70 |
num_steps = data.pop("num_steps", 25)
|
71 |
unconditional_guidance_scale = data.pop("unconditional_guidance_scale", 7.5)
|
|
|
62 |
context = base64.b64decode(contexts[0])
|
63 |
context = np.frombuffer(context, dtype="float32")
|
64 |
context = np.reshape(context, (batch_size, 77, 768))
|
65 |
+
print(context)
|
66 |
|
67 |
unconditional_context = base64.b64decode(contexts[1])
|
68 |
unconditional_context = np.frombuffer(unconditional_context, dtype="float32")
|
69 |
+
unconditional_context = np.reshape(unconditional_context, (batch_size, 77, 768))
|
70 |
+
print(unconditional_context)
|
71 |
|
72 |
num_steps = data.pop("num_steps", 25)
|
73 |
unconditional_guidance_scale = data.pop("unconditional_guidance_scale", 7.5)
|