diff --git a/dataset/dottxt_ai_outlines_task/task1706/Dockerfile b/dataset/dottxt_ai_outlines_task/task1706/Dockerfile index cac085e..bf85d86 100644 --- a/dataset/dottxt_ai_outlines_task/task1706/Dockerfile +++ b/dataset/dottxt_ai_outlines_task/task1706/Dockerfile @@ -26,7 +26,7 @@ ENV LLAMA_CPP_FORCE_CPU=1 ENV PYTORCH_ENABLE_MPS_FALLBACK=1 RUN uv pip install --system -e . && \ uv pip install --system pytest pytest-xdist pytest_mock pytest-asyncio pytest-benchmark pytest-cov && \ - uv pip install --system torch transformers sentencepiece xgrammar llama-cpp-python==0.3.16 psutil + uv pip install --system torch "transformers<5" "tokenizers<0.21" sentencepiece xgrammar llama-cpp-python==0.3.16 psutil # Copy the runner script COPY runner.sh /usr/local/bin/ diff --git a/dataset/dottxt_ai_outlines_task/task1706/runner.sh b/dataset/dottxt_ai_outlines_task/task1706/runner.sh index ef63deb..653367e 100644 --- a/dataset/dottxt_ai_outlines_task/task1706/runner.sh +++ b/dataset/dottxt_ai_outlines_task/task1706/runner.sh @@ -67,7 +67,7 @@ echo "Installing dependencies..." uv sync 2>/dev/null || true uv pip install --system -e . uv pip install --system pytest pytest-xdist pytest_mock pytest-asyncio pytest-benchmark pytest-cov -uv pip install --system torch transformers sentencepiece xgrammar llama-cpp-python==0.3.16 psutil +uv pip install --system torch "transformers<5" "tokenizers<0.21" sentencepiece xgrammar llama-cpp-python==0.3.16 psutil # Run tests with timeout echo "Running tests..."