From 0f9661823b62337e6067eadf63ddd76cb581e826 Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Wed, 14 Dec 2022 16:59:19 +0800
Subject: [PATCH] Update benchmark/run.py

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 benchmark/run.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/benchmark/run.py b/benchmark/run.py
index b048424..2016701 100644
--- 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