Spaces:
Runtime error
Runtime error
nguyenvulebinh
commited on
Commit
•
6d72d61
1
Parent(s):
66f50a6
Update infer.py
Browse files
infer.py
CHANGED
@@ -7,9 +7,7 @@ from model_handling import EncoderDecoderSpokenNorm
|
|
7 |
import os
|
8 |
import random
|
9 |
import data_handling
|
10 |
-
from transformers
|
11 |
-
from transformers.generation_stopping_criteria import StoppingCriteriaList
|
12 |
-
from transformers.generation_beam_search import BeamSearchScorer
|
13 |
from dataclasses import dataclass
|
14 |
from transformers.file_utils import ModelOutput
|
15 |
import utils
|
|
|
7 |
import os
|
8 |
import random
|
9 |
import data_handling
|
10 |
+
from transformers import LogitsProcessorList, StoppingCriteriaList, BeamSearchScorer
|
|
|
|
|
11 |
from dataclasses import dataclass
|
12 |
from transformers.file_utils import ModelOutput
|
13 |
import utils
|