xuxw98's picture
Update lit_llama/utils.py
c8ac827