* phi3

Co-authored-by: David Koski <dkoski@apple.com>
This commit is contained in:
Awni Hannun
2024-04-24 09:31:01 -07:00
committed by GitHub
parent 6c0b66f90a
commit b951b78eb2
7 changed files with 284 additions and 8 deletions

View File

@@ -30,6 +30,7 @@ public enum ModelType: String, Codable {
case mistral
case llama
case phi
case phi3
case gemma
case qwen2
case starcoder2
@@ -45,6 +46,10 @@ public enum ModelType: String, Codable {
let configuration = try JSONDecoder().decode(
PhiConfiguration.self, from: Data(contentsOf: configuration))
return PhiModel(configuration)
case .phi3:
let configuration = try JSONDecoder().decode(
Phi3Configuration.self, from: Data(contentsOf: configuration))
return Phi3Model(configuration)
case .gemma:
let configuration = try JSONDecoder().decode(
GemmaConfiguration.self, from: Data(contentsOf: configuration))