↓ 1 callersMethodprepare_vae_images(self, curr_kvlens, curr_rope, images, transforms, new_token_ids, timestep=0)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/bagel.py:423
↓ 1 callersMethodprepare_vae_latent(self, curr_kvlens, curr_rope, image_sizes, new_token_ids, generators=None)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/bagel.py:558
↓ 1 callersFunctionsave_ckpt(save_dir, transformer, global_step, accelerator, ema, transformer_trainable_parameters, config)
RL/flow_grpo/scripts/train_sd3_fast_guard.py:349
↓ 1 callersFunctionsave_ckpt(save_dir, transformer, global_step, accelerator, ema, transformer_trainable_parameters, config)
RL/flow_grpo/scripts/train_flux_kontext.py:315
↓ 1 callersFunctionsave_ckpt(save_dir, transformer, global_step, accelerator, ema, transformer_trainable_parameters, config)
RL/flow_grpo/scripts/train_sd3_GRPO_Guard.py:345
↓ 1 callersFunctionsave_ckpt(save_dir, transformer, global_step, accelerator, ema, transformer_trainable_parameters, config)
RL/flow_grpo/scripts/train_sd3_fast.py:344
↓ 1 callersFunctionsave_ckpt(save_dir, transformer, global_step, accelerator, ema, transformer_trainable_parameters, config)
RL/flow_grpo/scripts/train_flux_fast.py:339
↓ 1 callersFunctionsave_ckpt(save_dir, transformer, global_step, accelerator, ema, transformer_trainable_parameters, config)
RL/flow_grpo/scripts/train_sd3_sft.py:300
↓ 1 callersFunctionsave_ckpt(save_dir, transformer, global_step, accelerator, ema, transformer_trainable_parameters, config)
RL/flow_grpo/scripts/train_sd3.py:323