Methodpossible_correction_step(
self, euler_step, x, d, dt, next_sigma, denoiser, cond, uc
)
sgm/modules/diffusionmodules/sampling.py:217
Methodprepare_inputs_for_generation(
self, input_ids, past_key_values=None, attention_mask=None, inputs_embeds=None, **kwargs
)
llava/model/language_model/llava_llama.py:117
Methodsampler_step(self, sigma, next_sigma, denoiser, x, cond, uc, *args, **kwargs)
sgm/modules/diffusionmodules/sampling.py:79
Methodsampler_step(self, sigma, next_sigma, denoiser, x, cond, uc)
sgm/modules/diffusionmodules/sampling.py:244
Methodsampler_step(self, sigma, next_sigma, denoiser, x, cond, uc=None, **kwargs)
sgm/modules/diffusionmodules/sampling.py:268
Functionscaled_multihead_dot_product_attention(query, key, value, n_heads, past_key_value=None, softmax_scale=None, attn_bias=None, key_padding_mask=None, i
llava/model/language_model/mpt/attention.py:20
Functionstage2_process(input_image, prompt, a_prompt, n_prompt, num_samples, upscale, edm_steps, s_stage1, s_stage2,
gradio_demo_face.py:120
Functionstage2_process(input_image, prompt, a_prompt, n_prompt, num_samples, upscale, edm_steps, s_stage1, s_stage2,
gradio_demo_tiled.py:114
Functionstage2_process(input_image, prompt, a_prompt, n_prompt, num_samples, upscale, edm_steps, s_stage1, s_stage2,
gradio_demo.py:86
Functiontriton_flash_attn_fn(query, key, value, n_heads, past_key_value=None, softmax_scale=None, attn_bias=None, key_padding_mask=None, i
llava/model/language_model/mpt/attention.py:107
Functionxavier_normal_param_init_fn_(module: nn.Module, n_layers: int, d_model: Optional[int]=None, init_div_is_residual: Union[int, float, str, b
llava/model/language_model/mpt/param_init_fns.py:176
Functionxavier_uniform_param_init_fn_(module: nn.Module, n_layers: int, d_model: Optional[int]=None, init_div_is_residual: Union[int, float, str, b
llava/model/language_model/mpt/param_init_fns.py:169