Spaces:
Runtime error
Runtime error
Commit
•
932ed18
1
Parent(s):
fa97487
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
import os
|
| 2 |
os.environ["RWKV_JIT_ON"] = '1'
|
| 3 |
os.environ["RWKV_CUDA_ON"] = '1' # if '1' then use CUDA kernel for seq mode (much faster)
|
|
|
|
| 4 |
|
| 5 |
def install_cuda_toolkit():
|
| 6 |
CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
|
|
|
|
| 1 |
import os
|
| 2 |
os.environ["RWKV_JIT_ON"] = '1'
|
| 3 |
os.environ["RWKV_CUDA_ON"] = '1' # if '1' then use CUDA kernel for seq mode (much faster)
|
| 4 |
+
import subprocess
|
| 5 |
|
| 6 |
def install_cuda_toolkit():
|
| 7 |
CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
|