I assume they have special federal prisons, or sections, for felons who are at high risk. (Not that Trump is a felon.)
Use common sense, they are not going to put Trump in any kind of prison. He would probably be under house arrest.