ddpo-pytorch/setup.py
2023-09-15 22:54:51 -07:00

20 lines
426 B
Python

from setuptools import setup, find_packages
setup(
name="ddpo-pytorch",
version="0.0.1",
packages=["ddpo_pytorch"],
python_requires=">=3.10",
install_requires=[
"ml-collections",
"absl-py",
"diffusers[torch]==0.17.1",
"accelerate==0.17",
"wandb",
"torchvision",
"inflect==6.0.4",
"pydantic==1.10.9",
"transformers==4.30.2",
],
)