Spaces:
Runtime error
Runtime error
Commit
·
f8e14c1
1
Parent(s):
afbacf2
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,8 @@ def func(query):
|
|
4 |
try:
|
5 |
os.system("pip install rwkvstic inquirer transformers torch")
|
6 |
from rwkvstic.load import RWKV
|
7 |
-
model = RWKV("https://huggingface.co/BlinkDL/rwkv-4-pile-3b/resolve/main/RWKV-4-Pile-3B-Instruct-test1-20230124.pth"
|
|
|
8 |
|
9 |
def generate(query):
|
10 |
model.loadContext(newctx=query)
|
|
|
4 |
try:
|
5 |
os.system("pip install rwkvstic inquirer transformers torch")
|
6 |
from rwkvstic.load import RWKV
|
7 |
+
model = RWKV("https://huggingface.co/BlinkDL/rwkv-4-pile-3b/resolve/main/RWKV-4-Pile-3B-Instruct-test1-20230124.pth",
|
8 |
+
useGpu=False)
|
9 |
|
10 |
def generate(query):
|
11 |
model.loadContext(newctx=query)
|