| 290 | } |
| 291 | |
| 292 | LanguageModel::LanguageModel(const TextConfig &Config) : Config(Config) { |
| 293 | registerModule("model", std::make_shared<Gemma3Model>(Config)); |
| 294 | registerModule("lm_head", std::make_shared<nn::Linear>( |
| 295 | Config.HiddenSize, Config.VocabSize, false)); |
| 296 | } |
| 297 | |
| 298 | std::tuple<mx::array, std::optional<mx::array>> LanguageModel::forward( |
| 299 | const mx::array &Inputs, const std::optional<mx::array> &InputsEmbeds, |
nothing calls this directly
no test coverage detected