-
Notifications
You must be signed in to change notification settings - Fork 109
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Homogeneize generation params #428
base: main
Are you sure you want to change the base?
Conversation
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
…n the models to use other attributes later
071d502
to
286668f
Compare
Interestingly, using an explicit generation config seems to mess up this quite a lot |
…ams set by default which slow down generations
…ammatically if needed
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Great! I only reviewed it partially. Some comments below.
Co-authored-by: Nathan Habib <[email protected]> Co-authored-by: Albert Villanova del Moral <[email protected]>
Co-authored-by: Nathan Habib <[email protected]>
…ia the generationparams instead
Co-authored-by: Albert Villanova del Moral <[email protected]>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
nice to have a coherent config, few nits and good to go
@@ -124,6 +127,8 @@ class BaseModelConfig: | |||
model at a quantized precision. Needed for 4-bit and 8-bit precision. | |||
trust_remote_code (bool): Whether to trust remote code during model | |||
loading. | |||
generation_parameters (GenerationParameters): Range of parameters which will affect the generation. | |||
generation_config (GenerationConfig): GenerationConfig object (only passed during manual creation) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I don't understand why we have both generation paramaters and config ?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
If you want to manually create/load your own model (in Python, not using the CLI), you might want to provide it's generation_config directly instead of using the ligtheval generation parameters class
Co-authored-by: Albert Villanova del Moral <[email protected]> Co-authored-by: Nathan Habib <[email protected]>
Co-authored-by: Nathan Habib <[email protected]>
This PR does 3 things:
I would put
system_prompt
,fewshot_seeds
, anduse_chat_template
in the GenerationParameters too since they are generation parameters logically, but it can be another PRCloses #16