pytorch-labs / gpt-fast

Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
BSD 3-Clause "New" or "Revised" License
5.57k stars 507 forks source link

Minor fix for generate.py #117

Closed Artyom17 closed 7 months ago

Artyom17 commented 7 months ago

Super minor fix, the argument of the function was ignored.