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)
|