praeclarumjj3 commited on
Commit
a9e3b02
·
1 Parent(s): 6e254e5

Update chat.py

Browse files
Files changed (1) hide show
  1. chat.py +3 -3
chat.py CHANGED
@@ -168,21 +168,21 @@ class Chat:
168
  "text": server_error_msg,
169
  "error_code": 1,
170
  }
171
- yield json.dumps(ret).encode() + b"\0"
172
  except torch.cuda.CudaError as e:
173
  print("Caught torch.cuda.CudaError:", e)
174
  ret = {
175
  "text": server_error_msg,
176
  "error_code": 1,
177
  }
178
- yield json.dumps(ret).encode() + b"\0"
179
  except Exception as e:
180
  print("Caught Unknown Error", e)
181
  ret = {
182
  "text": server_error_msg,
183
  "error_code": 1,
184
  }
185
- yield json.dumps(ret).encode() + b"\0"
186
 
187
 
188
  if __name__ == "__main__":
 
168
  "text": server_error_msg,
169
  "error_code": 1,
170
  }
171
+ yield json.dumps(ret).encode()
172
  except torch.cuda.CudaError as e:
173
  print("Caught torch.cuda.CudaError:", e)
174
  ret = {
175
  "text": server_error_msg,
176
  "error_code": 1,
177
  }
178
+ yield json.dumps(ret).encode()
179
  except Exception as e:
180
  print("Caught Unknown Error", e)
181
  ret = {
182
  "text": server_error_msg,
183
  "error_code": 1,
184
  }
185
+ yield json.dumps(ret).encode()
186
 
187
 
188
  if __name__ == "__main__":