File size: 680 Bytes
7f48662 a713a09 7f48662 a713a09 7f48662 a713a09 7f48662 a713a09 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
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"
],
)
|