fix possibly OOM error

Former-commit-id: 0d590dffb41b0e832d9f87d20a23bcd0acd983aa
This commit is contained in:
hiyouga 2023-06-01 23:54:44 +08:00
parent 1512711ca2
commit 896dbfec16