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, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_qwenimage_fast_guard.py:124
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, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_flux_kontext.py:78
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, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_sd3_fast.py:98
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, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_sd3_sft.py:82
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, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_sd3.py:80
Method__init__(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_qwenimage.py:102
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, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_wan2_1.py:82
Method__init__(self, dataset, batch_size, k, num_replicas, rank, seed=0)
RL/flow_grpo/scripts/train_flux.py:84
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, model, vae_model, tokenizer, vae_transform, vit_transform, new_token_ids)
RL/flow_grpo/flow_grpo/bagel/inferencer.py:23
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