Commit 4e2ff6b5 authored by zhuwenwen's avatar zhuwenwen
Browse files

add step3-vl tuning

parent 71e4b817
...@@ -694,6 +694,11 @@ def main(args: argparse.Namespace): ...@@ -694,6 +694,11 @@ def main(args: argparse.Namespace):
topk = config.moe_topk[0] topk = config.moe_topk[0]
intermediate_size = config.moe_intermediate_size[0] intermediate_size = config.moe_intermediate_size[0]
shard_intermediate_size = 2 * intermediate_size // args.tp_size shard_intermediate_size = 2 * intermediate_size // args.tp_size
elif config.architectures[0] in ("Step3VLForConditionalGeneration"):
E = config.text_config.moe_num_experts
topk = config.text_config.moe_top_k
intermediate_size = config.text_config.moe_intermediate_size
shard_intermediate_size = 2 * intermediate_size // tp_size
else: else:
# Support for llama4 # Support for llama4
config = config.get_text_config() config = config.get_text_config()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment