Skip to content

Commit d13f3be

Browse files
authored
Add GPT OSS (#371)
* Add GPT OSS * extend createCausalMask to include lengths parameter * refactor: move YarnRoPE class to RoPEUtils.swift and clean up code
1 parent 3557601 commit d13f3be

File tree

6 files changed

+710
-4
lines changed

6 files changed

+710
-4
lines changed

Libraries/MLXLLM/LLMModelFactory.swift

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@ public class LLMTypeRegistry: ModelTypeRegistry, @unchecked Sendable {
5656
"lfm2": create(LFM2Configuration.self, LFM2Model.init),
5757
"baichuan_m1": create(BaichuanM1Configuration.self, BaichuanM1Model.init),
5858
"exaone4": create(Exaone4Configuration.self, Exaone4Model.init),
59+
"gpt_oss": create(GPTOSSConfiguration.self, GPTOSSModel.init),
5960
]
6061
}
6162
}

0 commit comments

Comments
 (0)