fix `llama` tests (#39161)

Commit

245 days ago

fix `llama` tests (#39161) * fix * fix * fix * fix * fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>