Update setup.py

82184857 · luopl · 09d54a38 · 82184857
Commit 82184857 authored May 08, 2025 by luopl
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 5 deletions

setup.py setup.py +5 -5

No files found.
--- a/setup.py
+++ b/setup.py
@@ -25,10 +25,10 @@ if __name__ == "__main__":
        author="Step-Video Team",
        packages=find_packages(),
        install_requires=[
-            "torchvision==0.18.1+das.opt1.dtk24042",
-            "torch==2.3.0+das.opt2.dtk24043",
-            "accelerate==1.0.0",
-            "transformers==4.39.1",
+            "torchvision==0.19.1+das.opt2.dtk2504",
+            "torch==2.4.1+das.opt2.dtk2504",
+            "accelerate==1.6.0",
+            "transformers==4.49.0",
            "diffusers==0.31.0",
            "sentencepiece==0.1.99",
            "imageio==2.37.0",
@@ -54,4 +54,4 @@ if __name__ == "__main__":
        ],
        include_package_data=True,
        python_requires=">=3.10",
-    )
\ No newline at end of file
+    )