axolotl-ai-cloud / axolotl

Go ahead and axolotl questions
https://axolotl-ai-cloud.github.io/axolotl/
Apache License 2.0
7.55k stars 816 forks source link

fix: parse model_kwargs #1825

Closed NanoCode012 closed 1 month ago

NanoCode012 commented 1 month ago

Description

This config was not parsed yet used within the code. This fixes it.

Motivation and Context

How has this been tested?

Yes, with Gemma-2-2b to set attn_implementation.

Screenshots (if appropriate)

Types of changes

Social Handles (Optional)