Unconditional Image Generation
Diffusers
Safetensors
PyTorch
DDPMPipeline
diffusion-models-class
medical-imaging
brain-mri
multiple-sclerosis
Instructions to use benetraco/brain_ddpm_64 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use benetraco/brain_ddpm_64 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("benetraco/brain_ddpm_64", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
End of training - Uploading full pipeline
Browse files
unet/diffusion_pytorch_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 454722660
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:be86090412d778434792ddfd01e8206b07f002ecac9cf1a7bca9fc2cc0ceb43a
|
| 3 |
size 454722660
|