format

Oneflow-Inc · Apr 10, 2024 · 2b46afe · 2b46afe
1 parent d0e03a5
commit 2b46afe
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/projects/Eval_LLM/main.py b/projects/Eval_LLM/main.py
@@ -4,12 +4,8 @@
 from transformers import AutoTokenizer as HF_AutoTokenizer
 
 import libai.utils.distributed as dist  # noqa
-from libai.config import LazyCall, LazyConfig
-from libai.models.utils.model_loader.base_loader import (  # noqa
-    ModelLoaderHuggerFace,
-    ModelLoaderLiBai,
-)
-from projects.Llama.llama import LlamaForCausalLM  # noqa
+from libai.config import LazyConfig
+from libai.models.utils.model_loader.base_loader import ModelLoaderLiBai # noqa
 
 
 class LLMLoaderLibai(ModelLoaderLiBai):