Fix prompt_tokens.shape[-1] issue with max_prompt_length 0e6c759 Running Surn commited on 25 days ago