train medusa stage-2

[smartliuhw]

how can i launch stage-2 training without using axolotl? can i just comment these lines in the train_code and save the whole model?

Medusa/medusa/train/train_legacy.py

Lines 346 to 348 in 5e98053

	# Freeze the base model
	for param in model.base_model.parameters():
	param.requires_grad = False

[smartliuhw]

and one more question. how can i obtain the medusa_choices during inference like the choices in this file? is it generated automatically or set manually?
looking forward to your relay!
@leeyeehoo @ctlllll