MCPcopy Create free account
hub / github.com/CompVis/diff2flow / diffusion.py

File diffusion.py

diff2flow/diffusion.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1import torch
2import einops
3import numpy as np
4import torch.nn as nn

Callers

nothing calls this directly

Calls 2

ForwardDiffusionClass · 0.85
q_sampleMethod · 0.45

Tested by

no test coverage detected