drlx

View on PyPIReverse Dependencies (0)

0.0.2 drlx-0.0.2-py3-none-any.whl

Wheel Details

Project: drlx
Version: 0.0.2
Filename: drlx-0.0.2-py3-none-any.whl
Download: [link]
Size: 29958
MD5: 7dcd951c8bc8f9b17281149915110eca
SHA256: 3db6a1c395832b6518898a4548753aceea502f314343124600cd14f57265ca13
Uploaded: 2023-08-18 21:25:44 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: drlx
Version: 0.0.2
Summary: DRLX is a library for distributed training of diffusion models via RL
Author: CarperAI
License: MIT
Classifier: Development Status :: 3 - Alpha
Classifier: Environment :: Console
Classifier: Intended Audience :: Science/Research
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: MIT License
Classifier: Natural Language :: English
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3 :: Only
Classifier: Programming Language :: Python :: Implementation :: CPython
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Typing :: Typed
Classifier: Operating System :: Unix
Requires-Python: >=3.9
Requires-Dist: torch
Requires-Dist: torchvision
Requires-Dist: torchtyping
Requires-Dist: einops
Requires-Dist: diffusers
Requires-Dist: transformers
Requires-Dist: accelerate
Requires-Dist: xformers
Requires-Dist: wandb
Requires-Dist: fastprogress
Requires-Dist: matplotlib
Requires-Dist: pygraphviz; extra == "benchmarks"
Requires-Dist: graphviz; extra == "benchmarks"
Requires-Dist: openai; extra == "benchmarks"
Requires-Dist: black; extra == "dev"
Requires-Dist: isort; extra == "dev"
Requires-Dist: flake8; extra == "dev"
Requires-Dist: flake8-pyproject; extra == "dev"
Requires-Dist: pydocstyle; extra == "dev"
Requires-Dist: mypy; extra == "dev"
Requires-Dist: pre-commit; extra == "dev"
Requires-Dist: pytest; extra == "dev"
Requires-Dist: pytest-cov; extra == "dev"
Requires-Dist: sphinx (==5.3.0); extra == "docs"
Requires-Dist: sphinx-rtd-theme; extra == "docs"
Requires-Dist: sphinx-autodoc-typehints; extra == "docs"
Requires-Dist: ipython; extra == "notebook"
Requires-Dist: swig (>=4.1.0); extra == "sodaracer"
Requires-Dist: box2d-py (==2.3.8); extra == "sodaracer"
Requires-Dist: pygame; extra == "sodaracer"
Requires-Dist: tritonclient[all]; extra == "triton"
Provides-Extra: benchmarks
Provides-Extra: dev
Provides-Extra: docs
Provides-Extra: notebook
Provides-Extra: sodaracer
Provides-Extra: triton
Description-Content-Type: text/markdown
[Description omitted; length: 1683 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.41.1)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
drlx/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
drlx/configs.py sha256=TghBGh2d_tEWRp1TXaCULe0JPCDJpQ4AGxjmS9-vFOw 11250
drlx/denoisers/__init__.py sha256=mvX5ZRIjwMHeT7A1SZgmZy_oX--kYulwwxBOO1gxEbg 3343
drlx/denoisers/ldm_unet.py sha256=zvaQVjpAdS7WzXoJf2pXmaGfkEvIKVLQMsG_wM9TvW0 5762
drlx/pipeline/__init__.py sha256=DKDKDxjIeQlaoo_wERwBxUZ4o6isF7129qCPNy7Xywg 2821
drlx/pipeline/imagenet_animal_prompts.py sha256=gLxes3BKYhG8iB1EqudMgJqvEsKuFuD2R0CviUsB9IM 1169
drlx/pipeline/pickapic_prompts.py sha256=VX9cqzB9dV_d5JMSclh1ZsrX0on4Uj6vj4C-fjT_QhY 549
drlx/reward_modelling/__init__.py sha256=glzP5Mn0Lej4ZpQ38jkxYUMbS-lb3PkqPnjazA_CCpE 2446
drlx/reward_modelling/aesthetics.py sha256=5ZlhA98WR3EDhcgb4JJoUFJMOML6OtE6bhuXMpu7bDY 2957
drlx/reward_modelling/pickscore.py sha256=fz1fdcLyKiKGF7sJYxX0fG9tpu2HgBwVAl28an7glPU 2411
drlx/reward_modelling/toy_rewards.py sha256=AiYiPbWAobYcu4qbr0dMUZQzxzGdbPwHdmC_r9azZtU 1536
drlx/sampling/__init__.py sha256=pANu3FIyVmFUqhgmClWhuhRMcx_VlsZw6v4vL4KmiqI 11248
drlx/trainer/__init__.py sha256=lRvGh5UDiqCBTrcf21Reiufx9oVWuOGjbom-ppSUglk 4577
drlx/trainer/ddpo_trainer.py sha256=pWH40dfMK4rlnhzvvjxJNgzSjKBdAFWBTwjfAOYS_8A 17054
drlx/trainer/ldm_trainer.py sha256=GPhJKeTQhlPxokENd9AUzcKtp1pM095Mk2FIyo2MKhU 5736
drlx/utils/__init__.py sha256=tsqg3pVeFbZgbJf9v1P6kyLYic54Sfmp-0DVQO-yf34 7254
drlx-0.0.2.dist-info/METADATA sha256=SYQsL50GjSgsU8laKnxwXYLjMcph5t3XJx8UkvQ6Ins 3965
drlx-0.0.2.dist-info/WHEEL sha256=5sUXSg9e4bi7lTLOHcm6QEYwO5TIF1TNbTSVFVjcJcc 92
drlx-0.0.2.dist-info/top_level.txt sha256=GIRDrcCivM05h4K-lQrVuJmxKwydH3BQVQE-JNTReQk 5
drlx-0.0.2.dist-info/RECORD

top_level.txt

drlx