transformers

4d797099 - ah actually we don't discard lm head if missing -> needs to be moved to correct device and etc