Spaces:

Hazzzardous
/

RWKV-Instruct

Sleeping

Hazzzardous commited on Feb 7, 2023

Commit

b74218d

1 Parent(s): 2432281

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,9 +3,9 @@ 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

 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.float32,
     useGPU=torch.cuda.is_available(),
+    dtype=torch.float32
 )
 import gradio as gr