update setup

2026-01-07 12:43:12 +00:00 · 2024-10-30 19:53:18 +08:00 · 2024-10-30 19:53:18 +08:00 · 1e8a7b0d50
commit 1e8a7b0d50
parent 0accd6278b
1 changed files with 1 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@ -16,7 +16,7 @@ setup(
    install_requires=[
        'torch>=1.6.0',
        'transformers==4.45.2',
-        'datasets>=2.19.0',
+        'datasets==2.19.0',
        'accelerate>=0.20.1',
        'sentence_transformers',
        'peft',