Hazzzardous commited on
Commit
f02c936
·
1 Parent(s): 7bca18c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -23,11 +23,7 @@ def to_md(text):
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
 
 
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