I had to do that and I ended up using FMLA. I don't think you can lose your job for receiving treatment. Just go through HR and you should be ok. I ended up taking a month off for IOP using leave and FMLA.

Look at it this way, your health is more important than a job. If you're not healthy, then how good will you be at your job?