├── .gitignore ├── LICENSE ├── README.md ├── arguments.py ├── assets ├── concept.png ├── example_prompts.txt ├── examples.png └── logo.png ├── environment.yml ├── main.py ├── models ├── RewardFlux.py ├── RewardPixart.py ├── RewardStableDiffusion.py ├── RewardStableDiffusionXL.py ├── __init__.py └── utils.py ├── rewards ├── __init__.py ├── aesthetic.py ├── base_reward.py ├── clip.py ├── hps.py ├── imagereward.py ├── pickscore.py └── utils.py └── training ├── __init__.py ├── optim.py └── trainer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/README.md -------------------------------------------------------------------------------- /arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/arguments.py -------------------------------------------------------------------------------- /assets/concept.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/assets/concept.png -------------------------------------------------------------------------------- /assets/example_prompts.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/assets/example_prompts.txt -------------------------------------------------------------------------------- /assets/examples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/assets/examples.png -------------------------------------------------------------------------------- /assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/assets/logo.png -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/environment.yml -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/main.py -------------------------------------------------------------------------------- /models/RewardFlux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/models/RewardFlux.py -------------------------------------------------------------------------------- /models/RewardPixart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/models/RewardPixart.py -------------------------------------------------------------------------------- /models/RewardStableDiffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/models/RewardStableDiffusion.py -------------------------------------------------------------------------------- /models/RewardStableDiffusionXL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/models/RewardStableDiffusionXL.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/models/utils.py -------------------------------------------------------------------------------- /rewards/__init__.py: -------------------------------------------------------------------------------- 1 | from .utils import clip_img_transform, get_reward_losses 2 | -------------------------------------------------------------------------------- /rewards/aesthetic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/rewards/aesthetic.py -------------------------------------------------------------------------------- /rewards/base_reward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/rewards/base_reward.py -------------------------------------------------------------------------------- /rewards/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/rewards/clip.py -------------------------------------------------------------------------------- /rewards/hps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/rewards/hps.py -------------------------------------------------------------------------------- /rewards/imagereward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/rewards/imagereward.py -------------------------------------------------------------------------------- /rewards/pickscore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/rewards/pickscore.py -------------------------------------------------------------------------------- /rewards/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/rewards/utils.py -------------------------------------------------------------------------------- /training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/training/__init__.py -------------------------------------------------------------------------------- /training/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/training/optim.py -------------------------------------------------------------------------------- /training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ReNO/HEAD/training/trainer.py --------------------------------------------------------------------------------