Skip to content

Conversation

@avesus
Copy link
Member

@avesus avesus commented Jun 22, 2024

A simple 4 layer generative Transformer with 16 attention heads.
One tweak of LLaMA is the embedding matrix, for which computed a pseudo-inverse used as unembedding that's being backpropagated.

After 50,000 iterations it generates wild tiny stories.

This PR adds an example of inference parameters and training (with example of run over 50,000 iterations).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant