Chris4K commited on
Commit
1661ecf
·
verified ·
1 Parent(s): e6fa4cf

Update services/chat_service.py

Browse files
Files changed (1) hide show
  1. services/chat_service.py +1 -1
services/chat_service.py CHANGED
@@ -233,7 +233,7 @@ class ChatService:
233
  early_stopping=False
234
  )
235
 
236
- response = tokenizer.decode(outputs[0][input_ids.shape[-1]:], skip_special_tokens=True)
237
 
238
 
239
  return response.strip()
 
233
  early_stopping=False
234
  )
235
 
236
+ response = self.tokenizer.decode(outputs[0][input_ids.shape[-1]:], skip_special_tokens=True)
237
 
238
 
239
  return response.strip()