Functionlayer_forward_default hidden_states: [batch, seq_len, hidden_size] mask: [(1, 1), seq_len, seq_len]
VDM_Decoder/sgm/models/sat_ours/transformer_defaults.py:277
Methodpossible_correction_step(self, euler_step, x, d, dt, next_sigma, denoiser, cond, uc)
VDM_Decoder/sgm/modules/diffusionmodules/sampling.py:231
Methodprepare_inputs_for_generation(
self, input_ids, past_key_values=None, attention_mask=None, inputs_embeds=None, **kwargs
)
MLLM/src/models/mistral/modeling_mistral.py:1162
Methodsampler_step(self, sigma, next_sigma, denoiser, x, cond, uc, *args, **kwargs)
VDM_Decoder/sgm/modules/diffusionmodules/sampling.py:78
Methodsampler_step(self, sigma, next_sigma, denoiser, x, cond, uc)
VDM_Decoder/sgm/modules/diffusionmodules/sampling.py:251
Methodsampler_step(self, sigma, next_sigma, denoiser, x, cond, uc=None, **kwargs)
VDM_Decoder/sgm/modules/diffusionmodules/sampling.py:275