2023-06-23 19:25:54 -07:00
|
|
|
from setuptools import setup, find_packages
|
|
|
|
|
|
|
|
|
|
setup(
|
2023-06-24 00:07:55 -07:00
|
|
|
name="ddpo-pytorch",
|
|
|
|
|
version="0.0.1",
|
2023-06-23 19:25:54 -07:00
|
|
|
packages=["ddpo_pytorch"],
|
2023-07-06 10:28:54 -07:00
|
|
|
python_requires=">=3.10",
|
2023-06-23 19:25:54 -07:00
|
|
|
install_requires=[
|
2023-06-24 00:07:55 -07:00
|
|
|
"ml-collections",
|
|
|
|
|
"absl-py",
|
|
|
|
|
"diffusers[torch]==0.17.1",
|
2023-09-15 22:54:51 -07:00
|
|
|
"accelerate==0.17",
|
2023-06-24 00:07:55 -07:00
|
|
|
"wandb",
|
|
|
|
|
"torchvision",
|
|
|
|
|
"inflect==6.0.4",
|
2023-07-04 00:40:42 -07:00
|
|
|
"pydantic==1.10.9",
|
2023-06-24 00:07:55 -07:00
|
|
|
"transformers==4.30.2",
|
2023-06-23 19:25:54 -07:00
|
|
|
],
|
2023-06-24 00:07:55 -07:00
|
|
|
)
|
