fix possibly OOM error

Former-commit-id: 0d590dffb41b0e832d9f87d20a23bcd0acd983aa
This commit is contained in:
hiyouga 2023-06-01 23:54:44 +08:00
parent 4edd01fc08
commit 7ef5821cba