chansung commited on
Commit
1b339ed
·
1 Parent(s): 5e4cdc9

Update handler.py

Browse files
Files changed (1) hide show
  1. 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)