FredZhang7 commited on
Commit
7faf5ad
·
verified ·
1 Parent(s): fd640d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,14 +1,14 @@
1
  import gradio as gr
2
  import gc, copy, re
3
- import urllib.request
4
  from rwkv.model import RWKV
5
  from rwkv.utils import PIPELINE, PIPELINE_ARGS
6
 
7
  ctx_limit = 4096
8
  title = "RWKV-5-World-0.1B-v1-20230803-ctx4096.pth"
9
 
10
- url = f"https://huggingface.co/BlinkDL/rwkv-5-world/resolve/main/{title}"
11
- urllib.request.urlretrieve(url, title)
 
12
 
13
  model = RWKV(model=title, strategy='cpu bf16')
14
  pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
 
1
  import gradio as gr
2
  import gc, copy, re
 
3
  from rwkv.model import RWKV
4
  from rwkv.utils import PIPELINE, PIPELINE_ARGS
5
 
6
  ctx_limit = 4096
7
  title = "RWKV-5-World-0.1B-v1-20230803-ctx4096.pth"
8
 
9
+ # import urllib.request
10
+ # url = f"https://huggingface.co/BlinkDL/rwkv-5-world/resolve/main/{title}"
11
+ # urllib.request.urlretrieve(url, title)
12
 
13
  model = RWKV(model=title, strategy='cpu bf16')
14
  pipeline = PIPELINE(model, "rwkv_vocab_v20230424")