Fix torch.compile on the Llama3.2 vision model

Fix torch.compile on the Llama3.2 vision model #1813