Function uniform_on_device

ldm/models/diffusion/ddpm.py:49–50 · view source on GitHub ↗

(r1, r2, shape, device)

Source from the content-addressed store, hash-verified

47
48
49	def uniform_on_device(r1, r2, shape, device):
50	return (r1 - r2) * torch.rand(*shape, device=device) + r2
51
52
53	class DDPM(pl.LightningModule):

nothing calls this directly

no outgoing calls

no test coverage detected