freeze black

eole-nlp · Jan 29, 2025 · 3cf7826 · 3cf7826
1 parent 1a8f969
commit 3cf7826
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/.github/workflows/push.yml b/.github/workflows/push.yml
@@ -25,7 +25,7 @@ jobs:
         pip install sacrebleu
         pip install flake8
         pip install rich
-        python -m pip install black flake8 pyproject-flake8
+        python -m pip install black<25 flake8 pyproject-flake8
         if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
     - name: Check code with Black
       run: |

diff --git a/recipes/pixtral/test_inference.py b/recipes/pixtral/test_inference.py
@@ -5,8 +5,7 @@
 from eole.inference_engine import InferenceEnginePY
 
 config = PredictConfig(
-    # model_path="./pixtral-12b",
-    model_path="./test-pixtral",
+    model_path="./pixtral-12b",
     src="dummy",
     max_length=500,
     gpu_ranks=[0],