Hugging Face's logo

Spaces:
BlinkDL
/
RWKV-Gradio-2
Running on T4

App Files Community
9

Use v12 7B

#6
by maccam912 - opened
base: refs/heads/main
←
from: refs/pr/6
Discussion Files changed
+1
-1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from pynvml import *
6
  nvmlInit()
7
  gpu_h = nvmlDeviceGetHandleByIndex(0)
8
  ctx_limit = 1536
9
- title = "RWKV-4-Raven-7B-v11x-Eng99%-Other1%-20230429-ctx8192"
10
 
11
  os.environ["RWKV_JIT_ON"] = '1'
12
  os.environ["RWKV_CUDA_ON"] = '1' # if '1' then use CUDA kernel for seq mode (much faster)
 
6
  nvmlInit()
7
  gpu_h = nvmlDeviceGetHandleByIndex(0)
8
  ctx_limit = 1536
9
+ title = "RWKV-4-Raven-7B-v12-Eng98%-Other2%-20230521-ctx8192"
10
 
11
  os.environ["RWKV_JIT_ON"] = '1'
12
  os.environ["RWKV_CUDA_ON"] = '1' # if '1' then use CUDA kernel for seq mode (much faster)