1. Added CUDA debugging support 2. Improved memory management during model inference. 3. Fixed logits output Test Device: Windows WSL Ubuntu 20.04 (python3.10 CUDA 5090 32G)