Text-to-Image
Diffusers
Safetensors
StableDiffusionPipeline
stable-diffusion
medical-imaging
fundus
compdiff
Instructions to use mahmoudibra98/compdiff-fundus with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use mahmoudibra98/compdiff-fundus with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("mahmoudibra98/compdiff-fundus", 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
- Local Apps Settings
- Draw Things
- DiffusionBee
Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -37,6 +37,22 @@ compdiff_pipeline.py # turnkey CompDiffPipeline (demographic-conditioned
|
|
| 37 |
training_config.yaml # full training configuration
|
| 38 |
```
|
| 39 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
## Quickstart (demographic-conditioned)
|
| 41 |
|
| 42 |
The bundled `compdiff_pipeline.py` reproduces the exact generation used in the paper —
|
|
|
|
| 37 |
training_config.yaml # full training configuration
|
| 38 |
```
|
| 39 |
|
| 40 |
+
## Requirements
|
| 41 |
+
|
| 42 |
+
```bash
|
| 43 |
+
pip install "diffusers>=0.35" transformers accelerate huggingface_hub safetensors pillow
|
| 44 |
+
```
|
| 45 |
+
|
| 46 |
+
Install **torch to match your CUDA driver** — a bare `pip install torch` may pull a build
|
| 47 |
+
newer than your driver supports (e.g. a `cu130` wheel on a CUDA 12.4 driver fails with
|
| 48 |
+
"NVIDIA driver too old" / `cuda available: False`). Pick the wheel for your CUDA version
|
| 49 |
+
from [pytorch.org](https://pytorch.org/get-started/locally/). Tested combo (A100, driver
|
| 50 |
+
550.x / CUDA 12.4):
|
| 51 |
+
|
| 52 |
+
```bash
|
| 53 |
+
pip install torch==2.6.0 --index-url https://download.pytorch.org/whl/cu124
|
| 54 |
+
```
|
| 55 |
+
|
| 56 |
## Quickstart (demographic-conditioned)
|
| 57 |
|
| 58 |
The bundled `compdiff_pipeline.py` reproduces the exact generation used in the paper —
|