Spaces:
Running
Running
Shreyas Meher commited on
Commit ·
18e47fb
1
Parent(s): d386d50
Remove torch from requirements.txt to preserve GPU installs
Browse files- README.md +11 -1
- requirements.txt +4 -1
README.md
CHANGED
|
@@ -107,7 +107,17 @@ On Windows, if you get a permission error, run PowerShell as Administrator and e
|
|
| 107 |
Set-ExecutionPolicy -ExecutionPolicy RemoteSigned -Scope LocalMachine
|
| 108 |
```
|
| 109 |
|
| 110 |
-
3. Install
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 111 |
|
| 112 |
```bash
|
| 113 |
pip install -r requirements.txt
|
|
|
|
| 107 |
Set-ExecutionPolicy -ExecutionPolicy RemoteSigned -Scope LocalMachine
|
| 108 |
```
|
| 109 |
|
| 110 |
+
3. Install PyTorch:
|
| 111 |
+
|
| 112 |
+
```bash
|
| 113 |
+
# CPU only (Mac, or no NVIDIA GPU):
|
| 114 |
+
pip install torch
|
| 115 |
+
|
| 116 |
+
# NVIDIA GPU (Windows/Linux):
|
| 117 |
+
pip install torch --index-url https://download.pytorch.org/whl/cu124
|
| 118 |
+
```
|
| 119 |
+
|
| 120 |
+
4. Install remaining dependencies:
|
| 121 |
|
| 122 |
```bash
|
| 123 |
pip install -r requirements.txt
|
requirements.txt
CHANGED
|
@@ -1,4 +1,7 @@
|
|
| 1 |
-
|
|
|
|
|
|
|
|
|
|
| 2 |
tensorflow
|
| 3 |
transformers>=4.40,<5
|
| 4 |
gradio
|
|
|
|
| 1 |
+
# Install PyTorch separately FIRST:
|
| 2 |
+
# CPU only: pip install torch
|
| 3 |
+
# NVIDIA GPU: pip install torch --index-url https://download.pytorch.org/whl/cu124
|
| 4 |
+
# Then: pip install -r requirements.txt
|
| 5 |
tensorflow
|
| 6 |
transformers>=4.40,<5
|
| 7 |
gradio
|