Tweaks for lm-eval-harness

[zphang]

• Only check for position_embedding_type if the field exists for the checkpoint-loaded args.
• Only load optimizer/lr scheduler states if user provide optimizer and lr scheduler.