Spaces:
Runtime error
Runtime error
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()
|