Spaces:
Running on T4
Running on T4
Commit ·
f9eb4f2
1
Parent(s): 45d682f
fix
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ from rwkv.utils import PIPELINE, PIPELINE_ARGS
|
|
| 9 |
import rwkv7_fast_v3a as v3a
|
| 10 |
|
| 11 |
nvmlInit()
|
| 12 |
-
|
| 13 |
|
| 14 |
ctx_limit = 7000
|
| 15 |
gen_limit = 1000
|
|
|
|
| 9 |
import rwkv7_fast_v3a as v3a
|
| 10 |
|
| 11 |
nvmlInit()
|
| 12 |
+
gpu_h = nvmlDeviceGetHandleByIndex(0)
|
| 13 |
|
| 14 |
ctx_limit = 7000
|
| 15 |
gen_limit = 1000
|