weiyi01191 commited on
Commit
398b172
·
1 Parent(s): 1da3c45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/llama2_test_config.yaml")
381
  parser.add_argument("--ckpt", type=str,
382
- default='checkpoints/video_llama_checkpoint_last.pth',
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数")