fix

huggingface · Feb 13, 2025 · 90099ba · 90099ba
1 parent e42032d
commit 90099ba
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.github/workflows/python-tests.yml b/.github/workflows/python-tests.yml
@@ -26,6 +26,7 @@ jobs:
           [
             "Repository only",
             "Everything else",
+            "Inference only"
 
           ]
         include:
@@ -64,7 +65,7 @@ jobs:
 
           case "${{ matrix.test_name }}" in
 
-            "Repository only" | "Everything else")
+            "Repository only" | "Everything else" | "Inference only")
               sudo apt update
               sudo apt install -y libsndfile1-dev
               ;;

diff --git a/tests/test_inference_client.py b/tests/test_inference_client.py
@@ -106,6 +106,7 @@
     },
 }
 
+CHAT_COMPLETION_MODEL = "HuggingFaceH4/zephyr-7b-beta"
 CHAT_COMPLETION_MESSAGES = [
     {"role": "system", "content": "You are a helpful assistant."},
     {"role": "user", "content": "What is deep learning?"},