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
|