Remove nn.Embedding layer from model size

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.67k stars 513 forks source link

Closed yanboliang closed 6 months ago