Fix model filename
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ desc = f'''链接:<a href='https://colab.research.google.com/drive/1J1gLMMMA8G
|
|
13 |
|
14 |
parser = argparse.ArgumentParser(prog = 'ChatGal RWKV')
|
15 |
parser.add_argument('--share',action='store_true')
|
16 |
-
parser.add_argument('--ckpt',type=str,default="rwkv-
|
17 |
parser.add_argument('--model_path',type=str,default=None,help="local model path")
|
18 |
parser.add_argument('--lora', type=str, default=None, help='lora checkpoint path')
|
19 |
parser.add_argument('--lora_alpha', type=float, default=0, help='lora alpha')
|
|
|
13 |
|
14 |
parser = argparse.ArgumentParser(prog = 'ChatGal RWKV')
|
15 |
parser.add_argument('--share',action='store_true')
|
16 |
+
parser.add_argument('--ckpt',type=str,default="rwkv-loramerge-0426-v2-4096-epoch11.pth")
|
17 |
parser.add_argument('--model_path',type=str,default=None,help="local model path")
|
18 |
parser.add_argument('--lora', type=str, default=None, help='lora checkpoint path')
|
19 |
parser.add_argument('--lora_alpha', type=float, default=0, help='lora alpha')
|