Spaces:

Synthia
/

ChatGal

Runtime error

wanicca commited on Apr 19, 2023

Commit

9848182

1 Parent(s): e0c64a2

update to epoch 20

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ desc = f'''链接：<a href='https://colab.research.google.com/drive/1J1gLMMMA8G
 parser = argparse.ArgumentParser(prog = 'ChatGal RWKV')
 parser.add_argument('--share',action='store_true')
-parser.add_argument('--ckpt',type=str,default="rwkv-chatgal-v1-3B-ctx4096-epoch10.pth")
 args = parser.parse_args()
 os.environ["RWKV_JIT_ON"] = '1'

 parser = argparse.ArgumentParser(prog = 'ChatGal RWKV')
 parser.add_argument('--share',action='store_true')
+parser.add_argument('--ckpt',type=str,default="rwkv-chatgal-v1-3B-ctx4096-epoch20.pth")
 args = parser.parse_args()
 os.environ["RWKV_JIT_ON"] = '1'