mahmoudibra98 commited on
Commit
dd91701
·
verified ·
1 Parent(s): d4c4055

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +16 -0
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 —