Spaces:
Running
Running
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数")
|