tch-rs
tch-rs copied to clipboard
Add no_grad to min gpt example
Add no_grad
to the sampling function to improve inference efficiency.