RWKV-7

Sleeping

Bytesculptor07 commited on Mar 13

Commit

849576a

verified ·

1 Parent(s): d5955ca

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import os, copy
 os.environ["CUDA_HOME"] = "/cuda/"
 os.environ["RWKV_V7_ON"] = '1'
 os.environ["RWKV_JIT_ON"] = '1'
-os.environ["RWKV_CUDA_ON"] = '1' # if '1' then use CUDA kernel for seq mode (much faster)
 from rwkv.model import RWKV

 os.environ["CUDA_HOME"] = "/cuda/"
 os.environ["RWKV_V7_ON"] = '1'
 os.environ["RWKV_JIT_ON"] = '1'
+os.environ["RWKV_CUDA_ON"] = '0' # if '1' then use CUDA kernel for seq mode (much faster)
 from rwkv.model import RWKV