Hazzzardous commited on
Commit
ab1c0b0
·
1 Parent(s): 28f6cd3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
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-1b5/resolve/main/RWKV-4-Pile-1B5-Instruct-test1-20230124.pth",
27
- "pytorch(cpu/gpu)",
28
- runtimedtype=torch.float32,
29
- useGPU=torch.cuda.is_available(),
30
- dtype=torch.float32
31
  )
32
  return model
33
 
@@ -233,7 +229,7 @@ chatiface = gr.Interface(
233
  demo = gr.TabbedInterface(
234
 
235
  [iface,chatiface],["Generative","Chatbot"],
236
- title="RWKV-4 (1.5b Instruct)",
237
 
238
  )
239
 
 
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
 
 
229
  demo = gr.TabbedInterface(
230
 
231
  [iface,chatiface],["Generative","Chatbot"],
232
+ title="RWKV-4 (7b Instruct, 8-Bit [cpu/for now])",
233
 
234
  )
235