Skip to content

GitLab

Commit 82184857 authored by luopl's avatar luopl
Browse files

Update setup.py

parent 09d54a38
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
setup.py
View file @ 82184857
...@@ -25,10 +25,10 @@ if __name__ == "__main__": ...@@ -25,10 +25,10 @@ if __name__ == "__main__":
author="Step-Video Team", author="Step-Video Team",
packages=find_packages(), packages=find_packages(),
install_requires=[ install_requires=[
"torchvision==0.18.1+das.opt1.dtk24042", "torchvision==0.19.1+das.opt2.dtk2504",
"torch==2.3.0+das.opt2.dtk24043", "torch==2.4.1+das.opt2.dtk2504",
"accelerate==1.0.0", "accelerate==1.6.0",
"transformers==4.39.1", "transformers==4.49.0",
"diffusers==0.31.0", "diffusers==0.31.0",
"sentencepiece==0.1.99", "sentencepiece==0.1.99",
"imageio==2.37.0", "imageio==2.37.0",
...@@ -54,4 +54,4 @@ if __name__ == "__main__": ...@@ -54,4 +54,4 @@ if __name__ == "__main__":
], ],
include_package_data=True, include_package_data=True,
python_requires=">=3.10", python_requires=">=3.10",
) )
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment