chore: isort

zhudotexe · Dec 9, 2024 · f860f42 · f860f42
1 parent 10a3b36
commit f860f42
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bench_engines.py b/bench_engines.py
@@ -88,10 +88,11 @@ def get_engine(model_class: str, model_id: str, context_size: int = None):
             )
     # ===== QWEN =====
     if model_class == "qwen":
-        from utils.qwen import QwenFunctionCallingAdapter
         from kani.ext.vllm import VLLMEngine
         from vllm import SamplingParams
 
+        from utils.qwen import QwenFunctionCallingAdapter
+
         if model_id == "Qwen/Qwen2.5-72B-Instruct":
             model = VLLMEngine(
                 model_id="Qwen/Qwen2.5-72B-Instruct",