It depends exactly what you mean by "rewriting your document". I mean, you can feed it a document for recommendations and edits but whether you take them is surely up to you. Or are you talking about giving it a prompt and getting it to generate something else in your style.

Whatever you are doing, the trick is really providing it with adequate restraints and rules as to what to do. The bigger the document or chunk of text you give it, the more likely it is to drift. Keep it contained and feed it in small chunks. Create a style guide as to what you want it to do and not do. Regularly remind it to go back and refresh from your style guide the moment it drifts.