├── .gitignore ├── LICENSE ├── README.md ├── assets ├── method.png └── performance.png ├── config ├── base.py └── nft.py ├── dataset ├── drawbench │ ├── test.txt │ └── train.txt ├── geneval │ ├── test_metadata.jsonl │ └── train_metadata.jsonl ├── ocr │ ├── test.txt │ └── train.txt └── pickscore │ ├── test.txt │ └── train.txt ├── flow_grpo ├── aesthetic_scorer.py ├── assets │ ├── activities.txt │ ├── activities_v0.txt │ ├── imagenet_classes.txt │ ├── object_names.txt │ ├── simple_animals.txt │ ├── simple_ocr_animals.txt │ ├── simple_ocr_animals_digit1.txt │ ├── simple_ocr_animals_digit3.txt │ └── simple_ocr_animals_digit5.txt ├── clip_scorer.py ├── diffusers_patch │ ├── pipeline_with_logprob.py │ ├── solver.py │ └── train_dreambooth_lora_sd3.py ├── ema.py ├── gen_eval.py ├── hpsv2_scorer.py ├── imagereward_scorer.py ├── ocr.py ├── pickscore_scorer.py ├── prompts.py ├── reward_ckpt_path.py ├── rewards.py ├── stat_tracking.py ├── test_cases │ ├── a photo of a brown giraffe and a white stop sign.png │ ├── cat.jpg │ ├── hello world.jpg │ └── nasa.jpg └── unifiedreward_scorer.py ├── scripts ├── evaluation.py └── train_nft_sd3.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/README.md -------------------------------------------------------------------------------- /assets/method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/assets/method.png -------------------------------------------------------------------------------- /assets/performance.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/assets/performance.png -------------------------------------------------------------------------------- /config/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/config/base.py -------------------------------------------------------------------------------- /config/nft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/config/nft.py -------------------------------------------------------------------------------- /dataset/drawbench/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/dataset/drawbench/test.txt -------------------------------------------------------------------------------- /dataset/drawbench/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/dataset/drawbench/train.txt -------------------------------------------------------------------------------- /dataset/geneval/test_metadata.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/dataset/geneval/test_metadata.jsonl -------------------------------------------------------------------------------- /dataset/geneval/train_metadata.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/dataset/geneval/train_metadata.jsonl -------------------------------------------------------------------------------- /dataset/ocr/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/dataset/ocr/test.txt -------------------------------------------------------------------------------- /dataset/ocr/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/dataset/ocr/train.txt -------------------------------------------------------------------------------- /dataset/pickscore/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/dataset/pickscore/test.txt -------------------------------------------------------------------------------- /dataset/pickscore/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/dataset/pickscore/train.txt -------------------------------------------------------------------------------- /flow_grpo/aesthetic_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/aesthetic_scorer.py -------------------------------------------------------------------------------- /flow_grpo/assets/activities.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/assets/activities.txt -------------------------------------------------------------------------------- /flow_grpo/assets/activities_v0.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/assets/activities_v0.txt -------------------------------------------------------------------------------- /flow_grpo/assets/imagenet_classes.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/assets/imagenet_classes.txt -------------------------------------------------------------------------------- /flow_grpo/assets/object_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/assets/object_names.txt -------------------------------------------------------------------------------- /flow_grpo/assets/simple_animals.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/assets/simple_animals.txt -------------------------------------------------------------------------------- /flow_grpo/assets/simple_ocr_animals.txt: -------------------------------------------------------------------------------- 1 | cat 2 | dog 3 | horse 4 | monkey 5 | rabbit -------------------------------------------------------------------------------- /flow_grpo/assets/simple_ocr_animals_digit1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/assets/simple_ocr_animals_digit1.txt -------------------------------------------------------------------------------- /flow_grpo/assets/simple_ocr_animals_digit3.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/assets/simple_ocr_animals_digit3.txt -------------------------------------------------------------------------------- /flow_grpo/assets/simple_ocr_animals_digit5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/assets/simple_ocr_animals_digit5.txt -------------------------------------------------------------------------------- /flow_grpo/clip_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/clip_scorer.py -------------------------------------------------------------------------------- /flow_grpo/diffusers_patch/pipeline_with_logprob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/diffusers_patch/pipeline_with_logprob.py -------------------------------------------------------------------------------- /flow_grpo/diffusers_patch/solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/diffusers_patch/solver.py -------------------------------------------------------------------------------- /flow_grpo/diffusers_patch/train_dreambooth_lora_sd3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/diffusers_patch/train_dreambooth_lora_sd3.py -------------------------------------------------------------------------------- /flow_grpo/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/ema.py -------------------------------------------------------------------------------- /flow_grpo/gen_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/gen_eval.py -------------------------------------------------------------------------------- /flow_grpo/hpsv2_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/hpsv2_scorer.py -------------------------------------------------------------------------------- /flow_grpo/imagereward_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/imagereward_scorer.py -------------------------------------------------------------------------------- /flow_grpo/ocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/ocr.py -------------------------------------------------------------------------------- /flow_grpo/pickscore_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/pickscore_scorer.py -------------------------------------------------------------------------------- /flow_grpo/prompts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/prompts.py -------------------------------------------------------------------------------- /flow_grpo/reward_ckpt_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/reward_ckpt_path.py -------------------------------------------------------------------------------- /flow_grpo/rewards.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/rewards.py -------------------------------------------------------------------------------- /flow_grpo/stat_tracking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/stat_tracking.py -------------------------------------------------------------------------------- /flow_grpo/test_cases/a photo of a brown giraffe and a white stop sign.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/test_cases/a photo of a brown giraffe and a white stop sign.png -------------------------------------------------------------------------------- /flow_grpo/test_cases/cat.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/test_cases/cat.jpg -------------------------------------------------------------------------------- /flow_grpo/test_cases/hello world.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/test_cases/hello world.jpg -------------------------------------------------------------------------------- /flow_grpo/test_cases/nasa.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/test_cases/nasa.jpg -------------------------------------------------------------------------------- /flow_grpo/unifiedreward_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/flow_grpo/unifiedreward_scorer.py -------------------------------------------------------------------------------- /scripts/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/scripts/evaluation.py -------------------------------------------------------------------------------- /scripts/train_nft_sd3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/scripts/train_nft_sd3.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/DiffusionNFT/HEAD/setup.py --------------------------------------------------------------------------------