Spaces:
Sleeping
Sleeping
Commit
·
398b172
1
Parent(s):
1da3c45
Update app.py
Browse files
app.py
CHANGED
|
@@ -377,9 +377,9 @@ def get_arguments():
|
|
| 377 |
"""获取参数配置"""
|
| 378 |
parser = argparse.ArgumentParser(description="MiniGPT4-Video参数")
|
| 379 |
parser.add_argument("--cfg-path", help="配置文件路径",
|
| 380 |
-
default="test_configs/
|
| 381 |
parser.add_argument("--ckpt", type=str,
|
| 382 |
-
default='checkpoints/
|
| 383 |
help="模型检查点路径")
|
| 384 |
parser.add_argument("--max_new_tokens", type=int, default=512,
|
| 385 |
help="最大生成token数")
|
|
|
|
| 377 |
"""获取参数配置"""
|
| 378 |
parser = argparse.ArgumentParser(description="MiniGPT4-Video参数")
|
| 379 |
parser.add_argument("--cfg-path", help="配置文件路径",
|
| 380 |
+
default="test_configs/mistral_test_config.yaml")
|
| 381 |
parser.add_argument("--ckpt", type=str,
|
| 382 |
+
default='checkpoints/video_mistral_checkpoint_last.pth',
|
| 383 |
help="模型检查点路径")
|
| 384 |
parser.add_argument("--max_new_tokens", type=int, default=512,
|
| 385 |
help="最大生成token数")
|