Accelerate reasoning during testing #16
Description
You said in the README that the code was optimized during the test to achieve a high inference speed, so lines 63,67 and 71 could be commented out. But why did I report an error that the size was incorrect after commenting them out