nguyenvulebinh commited on
Commit
6d72d61
1 Parent(s): 66f50a6

Update infer.py

Browse files
Files changed (1) hide show
  1. infer.py +1 -3
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.generation_logits_process import LogitsProcessorList
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