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