inoculatemedia commited on
Commit
75ea98f
·
verified ·
1 Parent(s): 6de1288

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -94,7 +94,7 @@ def _parse_args():
94
  parser.add_argument(
95
  "--ckpt_dir",
96
  type=str,
97
- default='./weights/Wan2.1-I2V-14B-480P',
98
  help="The path to the Wan checkpoint directory.")
99
  parser.add_argument(
100
  "--quant_dir",
@@ -104,12 +104,12 @@ def _parse_args():
104
  parser.add_argument(
105
  "--infinitetalk_dir",
106
  type=str,
107
- default='weights/InfiniteTalk/single/infinitetalk.safetensors',
108
  help="The path to the InfiniteTalk checkpoint directory.")
109
  parser.add_argument(
110
  "--wav2vec_dir",
111
  type=str,
112
- default='./weights/chinese-wav2vec2-base',
113
  help="The path to the wav2vec checkpoint directory.")
114
  parser.add_argument(
115
  "--dit_path",
 
94
  parser.add_argument(
95
  "--ckpt_dir",
96
  type=str,
97
+ default='Wan-AI/Wan2.1-I2V-14B-480P',
98
  help="The path to the Wan checkpoint directory.")
99
  parser.add_argument(
100
  "--quant_dir",
 
104
  parser.add_argument(
105
  "--infinitetalk_dir",
106
  type=str,
107
+ default='MeiGen-AI/InfiniteTalk/single/infinitetalk.safetensors',
108
  help="The path to the InfiniteTalk checkpoint directory.")
109
  parser.add_argument(
110
  "--wav2vec_dir",
111
  type=str,
112
+ default='TencentGameMate/chinese-wav2vec2-base',
113
  help="The path to the wav2vec checkpoint directory.")
114
  parser.add_argument(
115
  "--dit_path",