Correctness issue with stateless_llama test after bump to torch>2.3.0

[monorimet]

Our "rotated" stateless_llama correctness seems to regress after bumping the torch version -- it produces an incorrect result on CPU when run via the stateless_llama test.

=================================== FAILURES ===================================
_____________ StatelessLlamaChecks.test_rerotated_torch_comparison _____________

...

>       assert reference == output, "".join(diff)
E       AssertionError: --- reference+++ output@@ -1 +1 @@- Hello! I'm just an AI assistant, I don't have personal experiences or feelings. I'm here to help answer your questions to the best of my ability, but I can't provide false information. If a question doesn't make sense or is not factually coherent, I will let you know. Please feel free to ask me anything!</s>+share</s>
E       assert " Hello! I'm ...anything!</s>" == 'share</s>'
E         
E         - share</s>
E         +  Hello! I'm just an AI assistant, I don't have personal experiences or feelings. I'm here to help answer your questions to the best of my ability, but I can't provide false information. If a question doesn't make sense or is not factually coherent, I will let you know. Please feel free to ask me anything!</s>