G-Rost commited on
Commit
8c0e823
1 Parent(s): 172c734

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +18 -5
requirements.txt CHANGED
@@ -1,6 +1,19 @@
1
- accelerate
2
- diffusers
3
- invisible_watermark
4
  torch
5
- transformers
6
- xformers
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Core Libraries
 
 
2
  torch
3
+ diffusers>=0.19.3
4
+ transformers>=4.31.0
5
+ gradio
6
+
7
+ # Accelerate and Optimization (Optional for CPU)
8
+ # accelerate
9
+ # xformers
10
+
11
+ # Image Manipulation and Watermarking
12
+ Pillow
13
+ invisible-watermark
14
+
15
+ # (Optional) Alternative Model Dependencies
16
+ # scipy # Only needed if using LMSDiscreteScheduler for faster models
17
+
18
+ # Development Tools (Optional)
19
+ # jupyter # If running in a Jupyter Notebook