rename llama.cpp test as cpu

huggingface · Jul 25, 2024 · 0c45ae6 · 0c45ae6
1 parent d010916
commit 0c45ae6
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/test_cli_misc.yaml b/.github/workflows/test_cli_misc.yaml
@@ -54,4 +54,4 @@ jobs:
           pip install -e .[testing]
 
       - name: Run tests
-        run: pytest -s -k "cli and not (cpu or cuda or mps)"
+        run: pytest -s -k "cli and not (cpu or cuda)"
diff --git a/optimum_benchmark/task_utils.py b/optimum_benchmark/task_utils.py
@@ -69,7 +69,7 @@ def infer_library_from_model_name_or_path(model_name_or_path: str, revision: Opt
         model_info = huggingface_hub.model_info(model_name_or_path, revision=revision)
         inferred_library_name = getattr(model_info, "library_name", None)
 
-        if "gguf" in model_info.tags:
+        if inferred_library_name is None and "gguf" in model_info.tags:
             inferred_library_name = "llama_cpp"
 
         if inferred_library_name == "sentence-transformers":

diff --git a/...gs/mps_inference_llama_cpp_embedding.yaml → ...gs/cpu_inference_llama_cpp_embedding.yaml b/...gs/mps_inference_llama_cpp_embedding.yaml → ...gs/cpu_inference_llama_cpp_embedding.yaml
diff --git a/..._inference_llama_cpp_text_generation.yaml → ..._inference_llama_cpp_text_generation.yaml b/..._inference_llama_cpp_text_generation.yaml → ..._inference_llama_cpp_text_generation.yaml