Revert removing the unused imports (#385)

2024-04-23 22:36:33 +08:00
parent 9acc6e3504
commit 150d7020ed
33 changed files with 72 additions and 26 deletions
--- a/test/srt/model/reference_hf.py
+++ b/test/srt/model/reference_hf.py
@@ -1,4 +1,5 @@
 import argparse
+import os

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer