Skip to content

Commit 8710ec2

Browse files
require model-parallel in convert.py
1 parent 7c2466b commit 8710ec2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

inference/convert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def main(hf_ckpt_path, save_path, n_experts, mp):
7878
parser.add_argument("--hf-ckpt-path", type=str, required=True)
7979
parser.add_argument("--save-path", type=str, required=True)
8080
parser.add_argument("--n-experts", type=int, required=True)
81-
parser.add_argument("--model-parallel", type=int, default=1)
81+
parser.add_argument("--model-parallel", type=int, required=True)
8282
args = parser.parse_args()
8383
assert args.n_experts % args.model_parallel == 0
8484
main(args.hf_ckpt_path, args.save_path, args.n_experts, args.model_parallel)

0 commit comments

Comments
 (0)