nxphi47 commited on
Commit
57dd736
1 Parent(s): dc89849

Update multipurpose_chatbot/engines/transformers_engine.py

Browse files
multipurpose_chatbot/engines/transformers_engine.py CHANGED
@@ -437,14 +437,14 @@ class TransformersEngine(BaseEngine):
437
  do_sample=True,
438
  temperature=temperature,
439
  max_new_tokens=max_tokens,
440
- pad_token_id=self.processor.tokenizer.pad_token_id,
441
  )
442
 
443
  out_tokens = []
444
  response = None
445
  for token in generator:
446
  out_tokens.append(token.item())
447
- response = self.processor.tokenizer.decode(out_tokens)
448
  num_tokens += 1
449
  # print(f"{num_tokens=}", end='\r')
450
  # sys.stdout.flush()
 
437
  do_sample=True,
438
  temperature=temperature,
439
  max_new_tokens=max_tokens,
440
+ pad_token_id=self.tokenizer.pad_token_id,
441
  )
442
 
443
  out_tokens = []
444
  response = None
445
  for token in generator:
446
  out_tokens.append(token.item())
447
+ response = self.tokenizer.decode(out_tokens)
448
  num_tokens += 1
449
  # print(f"{num_tokens=}", end='\r')
450
  # sys.stdout.flush()