From 64f31d9ea96b0983e08194d5c6da9143171d1153 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Iv=C3=A1n=20Baldo?= Date: Wed, 13 Dec 2023 18:01:19 -0300 Subject: [PATCH] Forgot to import torch. --- hf/bench-bt_fa.py | 1 + 1 file changed, 1 insertion(+) diff --git a/hf/bench-bt_fa.py b/hf/bench-bt_fa.py index efd3357..005ee92 100644 --- a/hf/bench-bt_fa.py +++ b/hf/bench-bt_fa.py @@ -1,6 +1,7 @@ # Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM from optimum.bettertransformer import BetterTransformer +import torch import time import sys sys.path.append('../common/')