| 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='[email protected]', | |
| 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" | |
| ], | |
| ) | |