sd-polyp-128

作者 Amey-HH · 下载 9 · 点赞 0 · 更新 2023-11-04

pytorch diffusers unconditional-image-generation diffusion-models-class

Model Card for Unit 1 of the Diffusion Models Class 🧨

This model is a diffusion model for unconditional image generation of Polyps.

Usage

from diffusers import DDPMPipeline

pipeline = DDPMPipeline.from_pretrained('Amey-HH/sd-polyp-128')
image = pipeline().images[0]
image