Skip to content

Conversation

@JRosenkranz
Copy link
Collaborator

Added a variant for GPTBigCode 20B to load with get_model

@JRosenkranz JRosenkranz requested a review from nairbv April 11, 2024 13:23
@JRosenkranz JRosenkranz self-assigned this Apr 11, 2024
@JRosenkranz JRosenkranz requested a review from daviswer April 11, 2024 13:24

_llama_13b = {"emb_dim": 5120, "vocab_size": 32000, "n_predict": 3, "inner_dim": 4096}

_gpt_bigcode_20b = {
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

can we add a comment on what this config comes from (e.g. where a user might find appropriate weights for it or how we arrived at the set of configs, a paper reference if relevant, etc)

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.

2 participants