fix: 修复 MPS/CPU 设备上的内存泄漏问题#47

Open

chaye7417 wants to merge 1 commit intoElectricAlexis:mainfrom

chaye7417:fix/mps-memory-leak

Commits on Jan 31, 2026

fix: add @torch.no_grad() to prevent memory leak on MPS/CPU
chaye7417
committed