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.58k stars 508 forks source link

May I ask which version of PyTorch does this project correspond to? #36

Closed ye1024 closed 9 months ago

singingtower commented 10 months ago

same question

huntzhan commented 10 months ago

@ye1024 @singingtower You should use the nightly version as mentioned in README.

Chillee commented 10 months ago

Yeah nightly is needed for this repo.