| from setuptools import setup, find_packages |
|
|
| with open("README.md", mode="r", encoding="utf-8") as readme_file: |
| readme = readme_file.read() |
|
|
| setup( |
| name='OmniGen', |
| version='1.0.3', |
| description='OmniGen', |
| long_description=readme, |
| long_description_content_type="text/markdown", |
| author_email='2906698981@qq.com', |
| url='https://github.com/VectorSpaceLab/OmniGen', |
| packages=find_packages(), |
| include_package_data=True, |
| install_requires=[ |
| 'torch==2.3.1', |
| 'transformers==4.45.2', |
| 'datasets', |
| 'accelerate==0.26.1', |
| 'diffusers==0.30.3', |
| "timm", |
| "peft==0.9.0", |
| "safetensors" |
| ], |
| ) |
|
|
