Hugging Face's logo

Spaces:
johnpaulbin
/
translation-rwkv
Runtime error

App Files Community
johnpaulbin commited on
Commit
7f0c662
·
1 Parent(s): 6db75e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ nvmlInit()
7
  gpu_h = nvmlDeviceGetHandleByIndex(0)
8
  ctx_limit = 512
9
  title = "RWKV-4-World-0.4B-v1-20230529-ctx4096"
10
- os.environ["RWKV_JIT_ON"] = '1'
11
  os.environ["RWKV_CUDA_ON"] = '0' # if '1' then use CUDA kernel for seq mode (much faster)
12
 
13
  from rwkv.model import RWKV
 
7
  gpu_h = nvmlDeviceGetHandleByIndex(0)
8
  ctx_limit = 512
9
  title = "RWKV-4-World-0.4B-v1-20230529-ctx4096"
10
+ os.environ["RWKV_JIT_ON"] = '0'
11
  os.environ["RWKV_CUDA_ON"] = '0' # if '1' then use CUDA kernel for seq mode (much faster)
12
 
13
  from rwkv.model import RWKV