How do you treat a yeast infection while on antibiotics?

I believe I have a yeast infection because I have thick, white discharge and some itching. I’m on antibiotics which is making me prone to yeast. When I treated the yeast last week it went away just to come back this week. Do I need to wait until I’m done the antibiotics to treat it again? Worried it’ll just come right back.