fix training base model config

Btlmd · Btlmd · commit 68b50404b3d7 · 2023-09-17T13:17:37.000Z
diff --git a/medusa/train/train.py b/medusa/train/train.py
@@ -345,6 +345,7 @@ def train():
         model,
         medusa_num_heads=training_args.medusa_num_heads,
         medusa_num_layers=training_args.medusa_num_layers,
+        base_model_name_or_path=model_args.model_name_or_path,
     )
 
     # Format output dir

Original file line number	Diff line number	Diff line change
`@@ -345,6 +345,7 @@ def train():`
`345`	`345`	`model,`
`346`	`346`	`medusa_num_heads=training_args.medusa_num_heads,`
`347`	`347`	`medusa_num_layers=training_args.medusa_num_layers,`
	`348`	`+ base_model_name_or_path=model_args.model_name_or_path,`
`348`	`349`	`)`
`349`	`350`
`350`	`351`	`# Format output dir`