Spaces:
Sleeping
Sleeping
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.
|
441 |
)
|
442 |
|
443 |
out_tokens = []
|
444 |
response = None
|
445 |
for token in generator:
|
446 |
out_tokens.append(token.item())
|
447 |
-
response = self.
|
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()
|