MCPcopy Index your code
hub / github.com/CompVis/diff2flow / predict_eps_from_z_and_v

Method predict_eps_from_z_and_v

diff2flow/ddpm.py:281–285  ·  view source on GitHub ↗
(self, x_t, t, v)

Source from the content-addressed store, hash-verified

279 )
280
281 def predict_eps_from_z_and_v(self, x_t, t, v):
282 return (
283 extract_into_tensor(self.sqrt_alphas_cumprod, t, x_t.shape) * v +
284 extract_into_tensor(self.sqrt_one_minus_alphas_cumprod, t, x_t.shape) * x_t
285 )
286
287 def predict_eps_from_xstart(self, x_t, t, pred_xstart):
288 return (

Callers

nothing calls this directly

Calls 1

extract_into_tensorFunction · 0.70

Tested by

no test coverage detected