Hazzzardous commited on
Commit
c5301b2
·
1 Parent(s): 9f00cb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -23,7 +23,11 @@ def to_md(text):
23
  def get_model():
24
  model = None
25
  model = RWKV(
26
- "https://huggingface.co/Hazzzardous/RWKV-8Bit/resolve/main/RWKV-4-Pile-7B-Instruct.pqth",
 
 
 
 
27
  )
28
  return model
29
 
 
23
  def get_model():
24
  model = None
25
  model = RWKV(
26
+ "https://huggingface.co/BlinkDL/rwkv-4-pile-7b/resolve/main/RWKV-4-Pile-7B-Instruct-test1-20230124.pth",
27
+ "pytorch(cpu/gpu)",
28
+ runtimedtype=torch.bfloat16,
29
+ useGPU=torch.cuda.is_available(),
30
+ dtype=torch.bfloat16
31
  )
32
  return model
33