↓ 3 callersFunctionpipeline_with_logprob(
self,
prompt: Union[str, List[str]] = None,
negative_prompt: Union[str, List[str]] = None,
t
RL/flow_grpo/flow_grpo/diffusers_patch/qwenimage_pipeline_with_logprob.py:10
↓ 3 callersFunctionpipeline_with_logprob(
self,
prompt: Union[str, List[str]] = None,
prompt_2: Optional[Union[str, List[str]]] = None,
RL/flow_grpo/flow_grpo/diffusers_patch/flux_pipeline_with_logprob.py:22
↓ 3 callersFunctionscaled_dot_product_attention(
query,
key,
value,
attn_mask=None,
dropout_p=0.0,
is_c
eval/TextCrafter/TextCrafter_SD3/utils.py:172
↓ 2 callersMethod__init__(self, dim, num_attention_heads, attention_head_dim, qk_norm="rms_norm", eps=1e-6)
eval/TextCrafter/TextCrafter_Flux/textcrafter_transformer_flux.py:124
↓ 2 callersFunctioncompute_log_prob(transformer, pipeline, sample, j, embeds, pooled_embeds, config)
RL/flow_grpo/scripts/train_sd3_fast.py:199
↓ 2 callersFunctioncompute_log_prob(transformer, pipeline, sample, j, embeds, pooled_embeds, config)
RL/flow_grpo/scripts/train_sd3.py:181
↓ 2 callersFunctioncompute_log_prob(transformer, pipeline, sample, j, embeds, negative_embeds, config, **kwargs)
RL/flow_grpo/scripts/train_wan2_1.py:190
↓ 2 callersFunctionpipeline_with_logprob(
self,
prompt: Union[str, List[str]] = None,
prompt_2: Optional[Union[str, List[str]]] = None,
RL/flow_grpo/flow_grpo/diffusers_patch/sd3_pipeline_with_logprob_fast_guard.py:12
↓ 2 callersFunctionpipeline_with_logprob(
self,
prompt: Union[str, List[str]] = None,
prompt_2: Optional[Union[str, List[str]]] = None,
RL/flow_grpo/flow_grpo/diffusers_patch/flux_pipeline_with_logprob_fast_guard.py:23
↓ 2 callersFunctionpipeline_with_logprob(
self,
prompt: Union[str, List[str]] = None,
negative_prompt: Union[str, List[str]] = None,
t
RL/flow_grpo/flow_grpo/diffusers_patch/qwenimage_pipeline_with_logprob_fast_guard.py:10
↓ 2 callersFunctionpipeline_with_logprob(
self,
prompt: Union[str, List[str]] = None,
prompt_2: Optional[Union[str, List[str]]] = None,
RL/flow_grpo/flow_grpo/diffusers_patch/sd3_pipeline_with_logprob_fast.py:12
↓ 2 callersFunctionpipeline_with_logprob(
self,
prompt: Union[str, List[str]] = None,
negative_prompt: Union[str, List[str]] = None,
t
RL/flow_grpo/flow_grpo/diffusers_patch/qwenimage_pipeline_with_logprob_fast.py:10
↓ 2 callersFunctionpipeline_with_logprob(
self,
prompt: Union[str, List[str]] = None,
prompt_2: Optional[Union[str, List[str]]] = None,
RL/flow_grpo/flow_grpo/diffusers_patch/flux_pipeline_with_logprob_fast.py:23
↓ 2 callersMethodprepare_prompts(self, curr_kvlens, curr_rope, prompts, tokenizer, new_token_ids)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/bagel.py:239
↓ 2 callersMethodprepare_vit_images(self, curr_kvlens, curr_rope, images, transforms, new_token_ids)
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/bagel.py:305