MelodyGen Train Comment out inference method for traininng purpose Inference Comment out train for inference purpose