MCPcopy Create free account

hub / github.com/NVlabs/DiffusionNFT / functions

Functions148 in github.com/NVlabs/DiffusionNFT

⨍ Functions 148◇ Types & classes 18
Method__init__
(self, dtype, device)
flow_grpo/aesthetic_scorer.py:32
Method__init__
OCR reward calculator :param use_gpu: Whether to use GPU acceleration for PaddleOCR
flow_grpo/ocr.py:10
Method__init__
(self, device="cuda", dtype=torch.float32)
flow_grpo/imagereward_scorer.py:8
Method__init__
(self, device)
flow_grpo/clip_scorer.py:34
Method__init__
(self, device="cuda", dtype=torch.float32)
flow_grpo/pickscore_scorer.py:7
Method__init__
(self, image: Image.Image, objects)
flow_grpo/gen_eval.py:77
Method__iter__
(self)
scripts/train_nft_sd3.py:134
Method__len__
(self)
scripts/evaluation.py:66
Method__len__
(self)
scripts/evaluation.py:82
Method__len__
(self)
scripts/train_nft_sd3.py:85
Method__len__
(self)
scripts/train_nft_sd3.py:105
Method__len__
(self)
flow_grpo/gen_eval.py:86
Method__post_init__
(self)
flow_grpo/diffusers_patch/solver.py:194
Function_fn
(images, prompts, metadata)
flow_grpo/rewards.py:9
Functionaesthetic_score
(device)
flow_grpo/rewards.py:33
Functionclip_score
(device)
flow_grpo/rewards.py:50
Functioncollate_fn
(examples)
scripts/evaluation.py:89
Methodcollate_fn
(examples)
scripts/train_nft_sd3.py:92
Methodcollate_fn
(examples)
scripts/train_nft_sd3.py:112
Functioncounting
(nouns_file, low, high)
flow_grpo/prompts.py:68
Methodforward
(self, img)
flow_grpo/hpsv2_scorer.py:26
Methodforward
(self, tensor)
flow_grpo/hpsv2_scorer.py:47
Methodforward
(self, embed)
flow_grpo/aesthetic_scorer.py:27
Functiongeneral_ocr
()
flow_grpo/prompts.py:49
Functiongeneval_score
(device)
flow_grpo/rewards.py:113
Functionget_config
(name)
config/nft.py:7
Functionget_config
()
config/base.py:4
Functionhpsv2_score
(device)
flow_grpo/rewards.py:65
Functionimagenet_all
()
flow_grpo/prompts.py:33
Functionimagenet_animals
()
flow_grpo/prompts.py:37
Functionimagenet_dogs
()
flow_grpo/prompts.py:41
Functionimagereward_score
(device)
flow_grpo/rewards.py:96
Functionjpeg_compressibility
()
flow_grpo/rewards.py:23
Functionmain
(_)
scripts/train_nft_sd3.py:356
Functionnouns_activities
(nouns_file, activities_file)
flow_grpo/prompts.py:62
Functionocr_score
(device)
flow_grpo/rewards.py:163
Functionpickscore_score
(device)
flow_grpo/rewards.py:80
Functionsd3_geneval
()
config/nft.py:82
Functionsd3_hpsv2
()
config/nft.py:112
Functionsd3_multi_reward
()
config/nft.py:122
Functionsd3_ocr
()
config/nft.py:70
Functionsd3_pickscore
()
config/nft.py:97
Functionsimple_animals
()
flow_grpo/prompts.py:45
Functionsimple_ocr_animals
()
flow_grpo/prompts.py:53
Methodsync_with_model
Force the EMA parameters to be a direct copy of the given model parameters. This is used to create a snapshot for the rollout policy.
flow_grpo/ema.py:55
Functiontimed
(fn)
flow_grpo/gen_eval.py:24
Functionunifiedreward_score_sglang
(device)
flow_grpo/rewards.py:179
Functionwrapper
(*args, **kwargs)
flow_grpo/gen_eval.py:25
← previous101–148 of 148, ranked by callers