Fix (examples/llm): Fix infinite loop in LLM entrypoint with WikiText2

Fix (examples/llm): Fix infinite loop in LLM entrypoint with WikiText2 #994