Spaces:

Hazzzardous
/

RWKV-Instruct

Sleeping

Hazzzardous commited on Feb 7, 2023

Commit

2432281

1 Parent(s): 71942c1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,10 +1,10 @@
 from rwkvstic.load import RWKV
 import torch
 model = RWKV(
-    "https://huggingface.co/BlinkDL/rwkv-4-pile-3b/resolve/main/RWKV-4-Pile-3B-20221110-ctx4096.pth",
     "pytorch(cpu/gpu)",
     runtimedtype=torch.bfloat16,
-    useGPU=False,
     dtype=torch.bfloat16
 )
 import gradio as gr

 from rwkvstic.load import RWKV
 import torch
 model = RWKV(
+    "https://huggingface.co/BlinkDL/rwkv-4-pile-1b5/resolve/main/RWKV-4-Pile-1B5-Instruct-test1-20230124.pth",
     "pytorch(cpu/gpu)",
     runtimedtype=torch.bfloat16,
+    useGPU=torch.cuda.is_available(),
     dtype=torch.bfloat16
 )
 import gradio as gr