jamesaasher commited on
Commit
6a80a8e
·
verified ·
1 Parent(s): 5d34f66

Upload requirements.txt with huggingface_hub

Browse files
Files changed (1) hide show
  1. requirements.txt +22 -0
requirements.txt ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Core dependencies for diffusion models
2
+ torch>=2.0.0
3
+ torchvision>=0.15.0
4
+ matplotlib>=3.6.0
5
+ pillow>=9.2.0
6
+ numpy>=1.22.0
7
+ tqdm>=4.64.0
8
+
9
+ # Diffusers and transformers
10
+ diffusers>=0.18.0,<0.20.0
11
+ transformers>=4.30.0
12
+ accelerate>=0.20.0
13
+ safetensors>=0.3.1
14
+ datasets>=2.14.0 # For HuggingFace datasets (QuickDraw)
15
+
16
+ # Jupyter notebook support
17
+ ipython>=8.10.0
18
+ jupyter>=1.0.0
19
+ ipykernel>=6.0.0
20
+
21
+ # Optional: for high quality image processing
22
+ opencv-python>=4.6.0