Mohammed-Altaf commited on
Commit
f41aa83
1 Parent(s): bb10374

added accelerate to the requirements

Browse files
Files changed (1) hide show
  1. load_meta_data.py +1 -1
load_meta_data.py CHANGED
@@ -26,7 +26,7 @@ class ChatBot:
26
  # add extra parameters if models runs successfully
27
  )
28
  response = self.tokenizer.decode(outputs[0],skip_special_tokens=True)
29
- response = self.get_clean_response(response)
30
  return response
31
 
32
  def get_clean_response(self,response):
 
26
  # add extra parameters if models runs successfully
27
  )
28
  response = self.tokenizer.decode(outputs[0],skip_special_tokens=True)
29
+ # response = self.get_clean_response(response)
30
  return response
31
 
32
  def get_clean_response(self,response):