↓ 15 callersMethodprepare_latents(
self,
batch_size,
num_channels_latents,
height,
width,
eval/TextCrafter/TextCrafter_SD3/textcrafter_pipeline_sd3.py:717
↓ 14 callersMethodcheck_inputs(
self,
prompt,
prompt_2,
prompt_3,
height,
width,
eval/TextCrafter/TextCrafter_SD3/textcrafter_pipeline_sd3.py:621
↓ 10 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.py:11
↓ 6 callersFunctionencode_prompt(
text_encoders,
tokenizers,
prompt: str,
max_sequence_length,
device=None,
num_images
RL/flow_grpo/flow_grpo/diffusers_patch/train_dreambooth_lora_sd3.py:98
↓ 4 callersFunctioncompute_log_prob(transformer, pipeline, sample, j, embeds, pooled_embeds, config)
RL/flow_grpo/scripts/train_sd3_fast_guard.py:200
↓ 4 callersFunctioncompute_log_prob(transformer, pipeline, sample, j, config, rank)
RL/flow_grpo/scripts/train_qwenimage_fast_guard.py:226
↓ 4 callersFunctioncompute_log_prob(transformer, pipeline, sample, j, embeds, pooled_embeds, config)
RL/flow_grpo/scripts/train_sd3_GRPO_Guard.py:200
↓ 4 callersFunctionencode_prompt(
text_encoders,
tokenizers,
prompt: str,
max_sequence_length,
device=None,
num_images
RL/flow_grpo/flow_grpo/diffusers_patch/train_dreambooth_lora_flux.py:107
↓ 4 callersFunctionrepeat_kv This is the equivalent of torch.repeat_interleave(x, dim=1, repeats=n_rep). The hidden states go from (batch, num_key_value_heads, seqlen, he
RL/flow_grpo/flow_grpo/bagel/modeling/qwen2/modeling_qwen2.py:205
↓ 3 callersMethod_sde_step_with_logprob(
self,
model_output,
timestep,
prev_timestep,
d_timestep,
sam
RL/flow_grpo/flow_grpo/bagel/modeling/bagel/bagel.py:986
↓ 3 callersFunctiongenerate_rectangles_gurobi
Generates a rectangle corresponding to a given point, using the Gurobi.
Parameters:
- points: List of tuples [(x1, y1), (x2, y2), ..
eval/TextCrafter/TextCrafter_Flux/rectangles.py:9
↓ 3 callersFunctiongenerate_rectangles_gurobi
Generates a rectangle corresponding to a given point, using the Gurobi.
Parameters:
- points: List of tuples [(x1, y1), (x2, y2), ..
eval/TextCrafter/TextCrafter_SD3/rectangles.py:8