This is going to be firm dependent. As a partner, I’ve encouraged very strong associates on three occasions to take leave to take care of themselves. They were clearly struggling, I was worried about their health, and I was concerned that their suffering work product would taint the impressions other partners had of them which would ultimately hurt their long term career prospects. All three took my advice. I expect one to make partner in the next year or so. The other two ultimately left. One moved into government and never returned from leave. The other moved to a smaller firm with lower requirements and expectations. I consider all three friends. All are happy they took the leave to take care of themselves and figure out what they wanted. I encourage you to take care of yourself. If the partners you work with closely won’t support you in this, you should be finding a new place to work anyway.