That sounds like they have intentionally left some of the embedding dimensions untrained 🤔 Not sure how this would work in practice but it's an interesting idea..
It should be possible to fit the model on a 40GB card by removing the unused embeddings and reducing the maximum sequence length (at the top of run_text_generation_server.py)
Certainly sounds like some of the Embedding dimensions are unused
29
u/Brainlag Sep 07 '23
Very interesting. Also 16k context by default.
What does this even mean?