ModelListedExternal

Wsi Generator

A Diffusion Model for Generating WSI Patches

Sourcehuggingface.co/kaveh/wsi_generator/tree/6415238fa84cccb59e77104d955bc9d4d817d26d

KindOtherVersionRevision 6415238

Model source

Kind: Other
File size: 557 B–434 MB
Version: vRevision 6415238
Updated: Sep 26, 2026
Downloads: 9
Source: Hugging Face

Source description

WSI Generation with DDPM

WSI image

A Diffusion Model for Generating WSI Patches

How to use the model?

from diffusers import DiffusionPipeline

wsi_generator = DiffusionPipeline.from_pretrained("kaveh/wsi_generator")
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
wsi_generator.to(device)

generated_image = wsi_generator().images[0]
generated_image.save("wsi_generated.png")

there is also a docker image available for this model in the following link:

Sources

1 source

Hugging FacePrimaryVersion 6415238fa84cccb59e77104d955bc9d4d817d26d

huggingface.co/kaveh/wsi_generator/tree/6415238fa84cccb59e77104d955bc9d4d817d26d

Verified Sep 26

Model artifacts

12 artifacts

checkpoint-164000/scaler.pt

pt · 557 B · SHA-256 853cffeb6428…06ed · Hugging Face

Download

checkpoint-164000/unet_ema/diffusion_pytorch_model.bin

bin · 434 MB · SHA-256 c3f11816979e…52d7 · Hugging Face

Download

Source excerpts

2 excerpts

ReadmeSep 26huggingface.co/kaveh/wsi_generator

--- license: gpl-3.0 language: - en metrics: - mse pipeline_tag: unconditional-image-generation tags: - diffusion - image generation - unconditional - wsi --- # WSI Generation with DDPM ![WSI image](https://i.imgur.com/hCKTyfE.png) A Diffusion Model for Generating WSI Patches How to use the model? ```py from diffusers import DiffusionPipeline wsi_generator = DiffusionPipeline.from_pretrained("kaveh/wsi_generator") device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') wsi_generator.to(device) generated_image = wsi_generator().images[0] generated_image.save("wsi_generated.png") ``` there is also a docker image available for this model in the following link: [https://hub.docker.com/r/kaveh8/wsi-ddpm](https://hub.docker.com/r/kaveh8/wsi-ddpm)