hub / github.com/ImprintLab/MedSegDiff / dpm_solver_first_update

Method dpm_solver_first_update

guided_diffusion/dpm_solver.py:514–558 · view source on GitHub ↗

DPM-Solver-1 (equivalent to DDIM) from time `s` to time `t`. Args: x: A pytorch tensor. The initial value at time `s`. s: A pytorch tensor. The starting time, with the shape (1,). t: A pytorch tensor. The ending time, with the shape (1,).

(self, x, s, t, model_s=None, return_intermediate=False)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 3

singlestep_dpm_solver_updateMethod · 0.95

multistep_dpm_solver_updateMethod · 0.95

dpm_solver_adaptiveMethod · 0.95

Calls 4

model_fnMethod · 0.95

marginal_lambdaMethod · 0.80

marginal_log_mean_coeffMethod · 0.80

marginal_stdMethod · 0.80

Tested by

no test coverage detected