Merge branch 'hf-bench-bt_fa' into ibaldoall

ivanbaldo · Dec 13, 2023 · f23e133 · f23e133
2 parents f46e4f7 + 64f31d9
commit f23e133
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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/')