MCPcopy Create free account

hub / github.com/Shredded-Pork/TempFlow-GRPO / functions

Functions220 in github.com/Shredded-Pork/TempFlow-GRPO

⨍ Functions 220◇ Types & classes 28
Method__getitem__
(self, idx)
scripts/train_sd3_pr.py:75
Method__getitem__
(self, idx)
scripts/train_sd3.py:54
Method__getitem__
(self, idx)
scripts/train_sd3.py:73
Method__getitem__
(self, idx)
scripts/train_flux_pr.py:51
Method__getitem__
(self, idx)
scripts/train_flux_pr.py:70
Method__getitem__
(self, idx)
scripts/train_qwenimage.py:73
Method__getitem__
(self, idx)
scripts/train_qwenimage.py:92
Method__getitem__
(self, idx)
scripts/train_flux.py:50
Method__getitem__
(self, idx)
scripts/train_flux.py:69
Method__init__
(self, dataset, split='train')
scripts/train_qwenimage_pr.py:64
Method__init__
(self, dataset, split='train')
scripts/train_qwenimage_pr.py:85
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
scripts/train_qwenimage_pr.py:104
Method__init__
(self, dataset, split='train')
scripts/train_sd3_pr.py:48
Method__init__
(self, dataset, split='train')
scripts/train_sd3_pr.py:66
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
scripts/train_sd3_pr.py:85
Method__init__
(self, dataset, split='train')
scripts/train_sd3.py:46
Method__init__
(self, dataset, split='train')
scripts/train_sd3.py:64
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
scripts/train_sd3.py:83
Method__init__
(self, dataset, split='train')
scripts/train_flux_pr.py:43
Method__init__
(self, dataset, split='train')
scripts/train_flux_pr.py:61
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
scripts/train_flux_pr.py:80
Method__init__
(self, dataset, split='train')
scripts/train_qwenimage.py:62
Method__init__
(self, dataset, split='train')
scripts/train_qwenimage.py:83
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
scripts/train_qwenimage.py:102
Method__init__
(self, dataset, split='train')
scripts/train_flux.py:42
Method__init__
(self, dataset, split='train')
scripts/train_flux.py:60
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
scripts/train_flux.py:79
Method__init__
( self, sharding_strategy="FULL_SHARD", backward_prefetch="BACKWARD_PRE", cpu
flow_grpo/fsdp_utils.py:19
Method__init__
(self, optimizer)
flow_grpo/fsdp_utils.py:131
Method__init__
( self, parameters: Iterable[torch.nn.Parameter], decay: float = 0.9999,
flow_grpo/ema.py:9
Method__init__
(self, global_std=False)
flow_grpo/stat_tracking.py:6
Method__init__
(self, dtype)
flow_grpo/aesthetic_scorer.py:33
Method__init__
OCR reward calculator :param use_gpu: Whether to use GPU acceleration for PaddleOCR
flow_grpo/ocr.py:9
Method__init__
(self, device="cuda", dtype=torch.float32)
flow_grpo/imagereward_scorer.py:7
Method__init__
(self, device="cuda", dtype=torch.bfloat16)
flow_grpo/qwenvl.py:27
Method__init__
(self, device="cuda", dtype=torch.float32)
flow_grpo/pickscore_scorer.py:6
Method__iter__
(self)
scripts/train_qwenimage_pr.py:118
Method__iter__
(self)
scripts/train_sd3_pr.py:99
Method__iter__
(self)
scripts/train_sd3.py:97
Method__iter__
(self)
scripts/train_flux_pr.py:94
Method__iter__
(self)
scripts/train_qwenimage.py:116
Method__iter__
(self)
scripts/train_flux.py:93
Method__len__
(self)
scripts/train_qwenimage_pr.py:72
Method__len__
(self)
scripts/train_qwenimage_pr.py:91
Method__len__
(self)
scripts/train_sd3_pr.py:53
Method__len__
(self)
scripts/train_sd3_pr.py:72
Method__len__
(self)
scripts/train_sd3.py:51
Method__len__
(self)
scripts/train_sd3.py:70
Method__len__
(self)
scripts/train_flux_pr.py:48
Method__len__
(self)
scripts/train_flux_pr.py:67
Method__len__
(self)
scripts/train_qwenimage.py:70
Method__len__
(self)
scripts/train_qwenimage.py:89
Method__len__
(self)
scripts/train_flux.py:47
Method__len__
(self)
scripts/train_flux.py:66
Function_fn
(images, prompts, metadata)
flow_grpo/rewards.py:8
Functionaesthetic_score
()
flow_grpo/rewards.py:31
Methodcollate_fn
(examples)
scripts/train_qwenimage_pr.py:79
Methodcollate_fn
(examples)
scripts/train_qwenimage_pr.py:98
Methodcollate_fn
(examples)
scripts/train_sd3_pr.py:60
Methodcollate_fn
(examples)
scripts/train_sd3_pr.py:79
Methodcollate_fn
(examples)
scripts/train_sd3.py:58
Methodcollate_fn
(examples)
scripts/train_sd3.py:77
Methodcollate_fn
(examples)
scripts/train_flux_pr.py:55
Methodcollate_fn
(examples)
scripts/train_flux_pr.py:74
Methodcollate_fn
(examples)
scripts/train_qwenimage.py:77
Methodcollate_fn
(examples)
scripts/train_qwenimage.py:96
Methodcollate_fn
(examples)
scripts/train_flux.py:54
Methodcollate_fn
(examples)
scripts/train_flux.py:73
Functioncompute_text_embeddings
(prompt, text_encoders, tokenizers, max_sequence_length, device)
scripts/train_qwenimage_pr.py:148
Functioncompute_text_embeddings
(prompt, text_encoders, tokenizers, max_sequence_length, device)
scripts/train_qwenimage.py:146
Functioncounting
(nouns_file, low, high)
flow_grpo/prompts.py:64
Functioncreate_generator
(prompts, base_seed)
scripts/train_qwenimage_pr.py:194
Functioncreate_generator
(prompts, base_seed)
scripts/train_flux_pr.py:170
Functioncreate_generator
(prompts, base_seed)
scripts/train_qwenimage.py:192
Functioncreate_generator
(prompts, base_seed)
scripts/train_flux.py:169
Functiondeqa_score_remote
Submits images to DeQA and computes a reward.
flow_grpo/rewards.py:160
Functionflux
()
config/dgx.py:260
Functionflux_pr
()
config/dgx.py:309
Methodforward
(self, embed)
flow_grpo/aesthetic_scorer.py:28
Functiongeneral_ocr
()
flow_grpo/prompts.py:47
Functiongeneral_ocr_sd3
()
config/dgx.py:30
Functiongeneval_score
Submits images to GenEval and computes a reward.
flow_grpo/rewards.py:209
Functiongeneval_sd3
()
config/dgx.py:86
Functiongeneval_sd3_pr
()
config/dgx.py:128
Functionget_config
()
config/base.py:4
Functionget_config
(name)
config/dgx.py:434
Functionget_sigmas
(noise_scheduler, timesteps, accelerator, n_dim=4, dtype=torch.float32)
scripts/train_sd3_pr.py:178
Functionget_sigmas
(noise_scheduler, timesteps, accelerator, n_dim=4, dtype=torch.float32)
scripts/train_sd3.py:178
Functionget_transformer_layer_cls
()
scripts/train_qwenimage_pr.py:328
Functiongrad_checkpoint_check_fn
Check function to determine which modules to checkpoint
flow_grpo/fsdp_utils.py:75
Functionhpsv3_score_remote
(device)
flow_grpo/rewards.py:95
Functionimagenet_all
()
flow_grpo/prompts.py:32
Functionimagenet_animals
()
flow_grpo/prompts.py:36
Functionimagenet_dogs
()
flow_grpo/prompts.py:40
Functionimagereward_score
(device)
flow_grpo/rewards.py:62
Functionjpeg_compressibility
()
flow_grpo/rewards.py:22
Functionmain
(_)
scripts/train_qwenimage_pr.py:337
Functionmain
(_)
scripts/train_sd3_pr.py:341
Functionmain
(_)
scripts/train_sd3.py:342
Functionmain
(_)
scripts/train_flux_pr.py:322
← previousnext →101–200 of 220, ranked by callers