chainyo commited on
Commit
5fc3ac3
1 Parent(s): 1770058

fix pipeline import

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -19,7 +19,8 @@ from typing import Dict, List, Union
19
  from optimum.onnxruntime import ORTModelForSequenceClassification, ORTOptimizer, ORTQuantizer
20
  from optimum.onnxruntime.configuration import OptimizationConfig, AutoQuantizationConfig
21
  from optimum.pipelines import pipeline as ort_pipeline
22
- from transformers import BertTokenizer, BertForSequenceClassification, pt_pipeline
 
23
 
24
  from utils import calculate_inference_time
25
 
 
19
  from optimum.onnxruntime import ORTModelForSequenceClassification, ORTOptimizer, ORTQuantizer
20
  from optimum.onnxruntime.configuration import OptimizationConfig, AutoQuantizationConfig
21
  from optimum.pipelines import pipeline as ort_pipeline
22
+ from transformers import BertTokenizer, BertForSequenceClassification
23
+ from transformers import pipeline as pt_pipeline
24
 
25
  from utils import calculate_inference_time
26