Create requirements.txt
Browse files- requirements.txt +4 -0
requirements.txt
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
torch>=2.0.0
|
| 2 |
+
transformers>=4.35.0
|
| 3 |
+
gradio>=3.44.0
|
| 4 |
+
accelerate>=0.24.0 # Pour device_map="auto"
|