↓ 1 callersFunctionsave_ckpt(save_dir, transformer, global_step, accelerator, ema, transformer_trainable_parameters, config)
RL/flow_grpo/scripts/train_sd3_dpo.py:298
↓ 1 callersFunctionsave_ckpt(save_dir, transformer, global_step, accelerator, ema, transformer_trainable_parameters, config)
RL/flow_grpo/scripts/train_wan2_1.py:337
↓ 1 callersFunctionsave_ckpt(save_dir, transformer, global_step, accelerator, ema, transformer_trainable_parameters, config)
RL/flow_grpo/scripts/train_flux.py:315
↓ 1 callersFunctionsave_ckpt(save_dir, transformer, global_step, accelerator, ema, transformer_trainable_parameters, config)
RL/flow_grpo/scripts/train_flux_fast_guard.py:339
↓ 1 callersFunctionscaled_dot_product_attention(
query,
key,
value,
attn_mask=None,
dropout_p=0.0,
is_c
eval/TextCrafter/TextCrafter_Flux/utils.py:172