Skip to main content

TryOnDiffusion Inference

Guide to running inference with TryOnDiffusion.

Sampling​

from tryondiffusion.diffusion import Diffusion

diffusion = Diffusion(...)
# Load checkpoint
# ...

# Generate image
result = diffusion.sample(
    use_ema=True,
    conditional_inputs=(ic, jp, jg, ia)
)

See TryOnDiffusion README for complete inference guide.