zhangshuai/mlx-swift-examples
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add Gemma 2 (#88)

Browse Source
This commit is contained in:
Anthony DePasquale
2024-07-01 18:35:43 +02:00
committed by GitHub
parent 7957378077
commit 0c08f3a7e4
3 changed files with 125 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
Libraries/LLM/Configuration.swift
View File

@@ -32,6 +32,7 @@ public enum ModelType: String, Codable {
case phi
case phi3
case gemma
case gemma2
case qwen2
case starcoder2
case cohere
@@ -55,6 +56,10 @@ public enum ModelType: String, Codable {
let configuration = try JSONDecoder().decode(
GemmaConfiguration.self, from: Data(contentsOf: configuration))
return GemmaModel(configuration)
case .gemma2:
let configuration = try JSONDecoder().decode(
GemmaConfiguration.self, from: Data(contentsOf: configuration))
return Gemma2Model(configuration)
case .qwen2:
let configuration = try JSONDecoder().decode(
Qwen2Configuration.self, from: Data(contentsOf: configuration))