Function convert_model_output

flow_grpo/diffusers_patch/solver.py:263–272 · view source on GitHub ↗

(
    model_output,
    sample,
    sigmas,
    step_index,
)

Source from the content-addressed store, hash-verified

261
262
263	def convert_model_output(
264	model_output,
265	sample,
266	sigmas,
267	step_index,
268	) -> torch.Tensor:
269	sigma_t = sigmas[step_index]
270	x0_pred = sample - sigma_t * model_output
271
272	return x0_pred
273
274
275	def ddim_update(

ddim_stepFunction · 0.85

dpm_stepFunction · 0.85

no outgoing calls

no test coverage detected