Spaces:
Paused
Paused
Update utils.py
Browse files
utils.py
CHANGED
|
@@ -4,7 +4,7 @@ import logging
|
|
| 4 |
import json
|
| 5 |
import os
|
| 6 |
import datetime
|
| 7 |
-
import time
|
| 8 |
import hashlib
|
| 9 |
import csv
|
| 10 |
import requests
|
|
@@ -191,7 +191,6 @@ def greedy_search(input_ids: torch.Tensor,
|
|
| 191 |
|
| 192 |
generated_tokens.append(next_token[0].item())
|
| 193 |
text = tokenizer.decode(generated_tokens)
|
| 194 |
-
time.sleep(0.10)
|
| 195 |
yield text
|
| 196 |
if any([x in text for x in stop_words]):
|
| 197 |
del past_key_values
|
|
|
|
| 4 |
import json
|
| 5 |
import os
|
| 6 |
import datetime
|
| 7 |
+
#import time
|
| 8 |
import hashlib
|
| 9 |
import csv
|
| 10 |
import requests
|
|
|
|
| 191 |
|
| 192 |
generated_tokens.append(next_token[0].item())
|
| 193 |
text = tokenizer.decode(generated_tokens)
|
|
|
|
| 194 |
yield text
|
| 195 |
if any([x in text for x in stop_words]):
|
| 196 |
del past_key_values
|