MCPcopy Create free account

hub / github.com/CIawevy/TextPecker / functions

Functions1,395 in github.com/CIawevy/TextPecker

Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_sd3_fast_guard.py:99
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_qwenimage_fast_guard.py:61
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_qwenimage_fast_guard.py:105
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_qwenimage_fast_guard.py:124
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_qwenimage_fast.py:61
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_qwenimage_fast.py:105
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_qwenimage_fast.py:124
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_flux_kontext.py:47
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_flux_kontext.py:78
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_sd3_GRPO_Guard.py:43
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_sd3_GRPO_Guard.py:80
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_sd3_GRPO_Guard.py:99
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_sd3_fast.py:42
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_sd3_fast.py:79
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_sd3_fast.py:98
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_flux_fast.py:46
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_flux_fast.py:83
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_flux_fast.py:102
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_sd3_sft.py:45
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_sd3_sft.py:63
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_sd3_sft.py:82
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_qwenimage_edit.py:64
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_qwenimage_edit.py:96
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_sd3.py:43
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_sd3.py:61
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_sd3.py:80
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_qwenimage.py:62
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_qwenimage.py:83
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_qwenimage.py:102
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_sd3_dpo.py:43
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_sd3_dpo.py:61
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_sd3_dpo.py:80
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_wan2_1.py:45
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_wan2_1.py:63
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_wan2_1.py:82
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_flux.py:47
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_flux.py:65
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_flux.py:84
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_flux_fast_guard.py:46
Method__init__
(self, dataset, split='train')
RL/flow_grpo/scripts/train_flux_fast_guard.py:83
Method__init__
(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_flux_fast_guard.py:102
Method__init__
( self, sharding_strategy="FULL_SHARD", backward_prefetch="BACKWARD_PRE", cpu
RL/flow_grpo/flow_grpo/fsdp_utils.py:19
Method__init__
(self)
RL/flow_grpo/flow_grpo/fsdp_utils.py:109
Method__init__
( self, parameters: Iterable[torch.nn.Parameter], decay: float = 0.9999,
RL/flow_grpo/flow_grpo/ema.py:9
Method__init__
(self, global_std=False)
RL/flow_grpo/flow_grpo/stat_tracking.py:6
Method__init__
(self, dtype)
RL/flow_grpo/flow_grpo/aesthetic_scorer.py:33
Method__init__
OCR reward calculator :param use_gpu: Whether to use GPU acceleration for PaddleOCR :param custom_model_dir: Custom directory
RL/flow_grpo/flow_grpo/ocr.py:134
Method__init__
OCR reward calculator :param use_gpu: Whether to use GPU acceleration for PaddleOCR :param custom_model_dir: Custom directory
RL/flow_grpo/flow_grpo/ocr.py:219
Method__init__
OCR reward calculator :param use_gpu: Whether to use GPU acceleration for PaddleOCR :param custom_model_dir: Custom directory
RL/flow_grpo/flow_grpo/ocr.py:352
Method__init__
OCR reward calculator :param use_gpu: Whether to use GPU acceleration for PaddleOCR
RL/flow_grpo/flow_grpo/ocr.py:462
Method__init__
(self, device="cuda", dtype=torch.float32)
RL/flow_grpo/flow_grpo/imagereward_scorer.py:7
Method__init__
(self, device)
RL/flow_grpo/flow_grpo/clip_scorer.py:31
Method__init__
(self, device="cuda", dtype=torch.bfloat16)
RL/flow_grpo/flow_grpo/qwenvl.py:27
Method__init__
(self, device="cuda", dtype=torch.float32)
RL/flow_grpo/flow_grpo/pickscore_scorer.py:6
Method__init__
(self, model, vae_model, tokenizer, vae_transform, vit_transform, new_token_ids)
RL/flow_grpo/flow_grpo/bagel/inferencer.py:23
Method__init__
(self, in_channels: int)
RL/flow_grpo/flow_grpo/bagel/modeling/autoencoder.py:39
Method__init__
(self, in_channels: int, out_channels: int)
RL/flow_grpo/flow_grpo/bagel/modeling/autoencoder.py:69
Method__init__
(self, in_channels: int)
RL/flow_grpo/flow_grpo/bagel/modeling/autoencoder.py:99
Method__init__
(self, in_channels: int)
RL/flow_grpo/flow_grpo/bagel/modeling/autoencoder.py:112
Method__init__
( self, ch: int, out_ch: int, ch_mult: list[int], num_res_blocks: int,
RL/flow_grpo/flow_grpo/bagel/modeling/autoencoder.py:197
Method__init__
(self, sample: bool = True, chunk_dim: int = 1)
RL/flow_grpo/flow_grpo/bagel/modeling/autoencoder.py:276
Method__init__
(self, params: AutoEncoderParams)
RL/flow_grpo/flow_grpo/bagel/modeling/autoencoder.py:291
Method__init__
Qwen2RMSNorm is equivalent to T5LayerNorm
RL/flow_grpo/flow_grpo/bagel/modeling/qwen2/modeling_qwen2.py:46
Method__init__
( self, dim=None, max_position_embeddings=2048, base=10000, device=Non
RL/flow_grpo/flow_grpo/bagel/modeling/qwen2/modeling_qwen2.py:67
Method__init__
(self, config: Qwen2Config, layer_idx: Optional[int] = None)
RL/flow_grpo/flow_grpo/bagel/modeling/qwen2/modeling_qwen2.py:223
Method__init__
(self, *args, **kwargs)
RL/flow_grpo/flow_grpo/bagel/modeling/qwen2/modeling_qwen2.py:332
Method__init__
(self, config: Qwen2Config, layer_idx: int)
RL/flow_grpo/flow_grpo/bagel/modeling/qwen2/modeling_qwen2.py:448
Method__init__
(self, config: Qwen2Config)
RL/flow_grpo/flow_grpo/bagel/modeling/qwen2/modeling_qwen2.py:662
Method__init__
(self, config)
RL/flow_grpo/flow_grpo/bagel/modeling/qwen2/modeling_qwen2.py:815
Method__init__
( self, vocab_file=None, merges_file=None, tokenizer_file=None, unk_to
RL/flow_grpo/flow_grpo/bagel/modeling/qwen2/tokenization_qwen2_fast.py:72
Method__init__
( self, vocab_size=151936, hidden_size=4096, intermediate_size=22016,
RL/flow_grpo/flow_grpo/bagel/modeling/qwen2/configuration_qwen2.py:121
Method__init__
( self, vocab_file, merges_file, errors="replace", unk_token="<|endoft
RL/flow_grpo/flow_grpo/bagel/modeling/qwen2/tokenization_qwen2.py:126
Method__init__
(self, hidden_size, frequency_embedding_size=256)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/modeling_utils.py:78
Method__init__
(self, max_num_patch_per_side, hidden_size)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/modeling_utils.py:128
Method__init__
( self, hidden_size=768, intermediate_size=3072, num_hidden_layers=12,
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/siglip_navit.py:71
Method__init__
(self, dim, max_h, max_w, base=10000)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/siglip_navit.py:103
Method__init__
(self, config: SiglipVisionConfig)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/siglip_navit.py:146
Method__init__
(self, *args, **kwargs)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/siglip_navit.py:199
Method__init__
(self, config: SiglipVisionConfig)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/siglip_navit.py:263
Method__init__
(self, config: SiglipVisionConfig)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/siglip_navit.py:304
Method__init__
(self, config: SiglipVisionConfig)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/siglip_navit.py:331
Method__init__
(self, config: SiglipVisionConfig)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/siglip_navit.py:378
Method__init__
( self, vocab_size=151936, hidden_size=4096, intermediate_size=22016,
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/qwen2_navit.py:149
Method__init__
(self, config, layer_idx: Optional[int] = None)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/qwen2_navit.py:234
Method__init__
(self, config, layer_idx: Optional[int] = None)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/qwen2_navit.py:379
Method__init__
(self, config, layer_idx: Optional[int] = None)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/qwen2_navit.py:606
Method__init__
( self, config, layer_idx: Optional[int] = None, attn_module: Optional[Qwen
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/qwen2_navit.py:690
Method__init__
(self, config, layer_idx: Optional[int] = None)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/qwen2_navit.py:822
Method__init__
(self, config)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/qwen2_navit.py:931
Method__init__
(self, config)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/qwen2_navit.py:1072
Method__init__
( self, visual_gen=True, visual_und=True, llm_config=None, vit_config=
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/bagel.py:34
Method__init__
( self, vocab_file, eos_token="</s>", unk_token="<unk>", pad_token="</
RL/flow_grpo/flow_grpo/bagel/modeling/siglip/tokenization_siglip.py:77
Method__init__
( self, vocab_size=32000, hidden_size=768, intermediate_size=3072, num
RL/flow_grpo/flow_grpo/bagel/modeling/siglip/configuration_siglip.py:72
Method__init__
( self, hidden_size=768, intermediate_size=3072, num_hidden_layers=12,
RL/flow_grpo/flow_grpo/bagel/modeling/siglip/configuration_siglip.py:171
Method__init__
(self, config: SiglipVisionConfig)
RL/flow_grpo/flow_grpo/bagel/modeling/siglip/modeling_siglip.py:240
Method__init__
(self, config: SiglipTextConfig)
RL/flow_grpo/flow_grpo/bagel/modeling/siglip/modeling_siglip.py:312
Method__init__
(self, config)
RL/flow_grpo/flow_grpo/bagel/modeling/siglip/modeling_siglip.py:348
Method__init__
(self, *args, **kwargs)
RL/flow_grpo/flow_grpo/bagel/modeling/siglip/modeling_siglip.py:430
Method__init__
(self, config: SiglipConfig)
RL/flow_grpo/flow_grpo/bagel/modeling/siglip/modeling_siglip.py:604
Method__init__
(self, config: SiglipConfig)
RL/flow_grpo/flow_grpo/bagel/modeling/siglip/modeling_siglip.py:833
← previousnext →801–900 of 1,395, ranked by callers