Lock vllm version to avoid bug

2025-04-07 11:57:15 +02:00
parent cda067eb07
commit faf78b7948
2 changed files with 1 additions and 283 deletions
--- a/tools/train/setup.py
+++ b/tools/train/setup.py
@@ -28,7 +28,7 @@ setup(
        'transformers>=4.30.0',
        'trl>=0.7.8',
        'unsloth>=2025.3',
-        'vllm>=0.8',
+        'vllm==0.8.2',
    ],
    python_requires='>=3.10',
 )