Update README.md
Browse files
README.md
CHANGED
|
@@ -26,13 +26,13 @@ tags:
|
|
| 26 |
### 📦 Install Dependencies
|
| 27 |
#### 🟢 For Nvidia CUDA
|
| 28 |
```bash
|
| 29 |
-
pip install torch rwkv gradio
|
| 30 |
```
|
| 31 |
#### 🔴 For AMD ROCm
|
| 32 |
- set ```os.environ["RWKV_CUDA_ON"] = '0' ```
|
| 33 |
```bash
|
| 34 |
pip install torch --index-url https://download.pytorch.org/whl/rocm6.3
|
| 35 |
-
pip install rwkv
|
| 36 |
```
|
| 37 |
|
| 38 |
### 😜 Run The demo
|
|
|
|
| 26 |
### 📦 Install Dependencies
|
| 27 |
#### 🟢 For Nvidia CUDA
|
| 28 |
```bash
|
| 29 |
+
pip install torch rwkv gradio
|
| 30 |
```
|
| 31 |
#### 🔴 For AMD ROCm
|
| 32 |
- set ```os.environ["RWKV_CUDA_ON"] = '0' ```
|
| 33 |
```bash
|
| 34 |
pip install torch --index-url https://download.pytorch.org/whl/rocm6.3
|
| 35 |
+
pip install rwkv gradio
|
| 36 |
```
|
| 37 |
|
| 38 |
### 😜 Run The demo
|