Update benchmark/run.py

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · 0f9661823b
1 changed files with 1 additions and 0 deletions
--- a/benchmark/run.py
+++ b/benchmark/run.py
@ -1,4 +1,5 @@
 import os
+import torch
 import onnxruntime
 from transformers.onnx.features import FeaturesManager
 from transformers.onnx import validate_model_outputs