.gitignore
README.md
pyproject.toml
requirements.txt
configs/ddpo_sd.yml
docs/Makefile
docs/make.bat
docs/source/conf.py
docs/source/denoisers.rst
docs/source/example.rst
docs/source/index.rst
docs/source/installation.rst
docs/source/pipeline.rst
docs/source/reward_modelling.rst
docs/source/sampling.rst
docs/source/trainer.rst
examples/contrasted_panda_inference.py
examples/contrasted_panda_training.py
src/drlx/__init__.py
src/drlx/configs.py
src/drlx.egg-info/PKG-INFO
src/drlx.egg-info/SOURCES.txt
src/drlx.egg-info/dependency_links.txt
src/drlx.egg-info/requires.txt
src/drlx.egg-info/top_level.txt
src/drlx/denoisers/__init__.py
src/drlx/denoisers/ldm_unet.py
src/drlx/pipeline/__init__.py
src/drlx/pipeline/imagenet_animal_prompts.py
src/drlx/pipeline/pickapic_prompts.py
src/drlx/reward_modelling/__init__.py
src/drlx/reward_modelling/aesthetics.py
src/drlx/reward_modelling/pickscore.py
src/drlx/reward_modelling/toy_rewards.py
src/drlx/sampling/__init__.py
src/drlx/trainer/__init__.py
src/drlx/trainer/ddpo_trainer.py
src/drlx/trainer/ldm_trainer.py
src/drlx/utils/__init__.py
tests/accelerate_checkpoint_test.py
tests/ddpo_unet_pipeline_test.py
tests/test_ddpo_trainer.py
tests/test_ddpo_trainer_imagenet.py