Some fixes for gemma2 (#99)

* some fixes for gemma2 * format * fixes * format
2024-08-01 20:06:11 -07:00
parent ac6bdfccec
commit 885e520ecd
5 changed files with 326 additions and 110 deletions
--- a/Libraries/LLM/Configuration.swift
+++ b/Libraries/LLM/Configuration.swift
@@ -58,7 +58,7 @@ public enum ModelType: String, Codable {
            return GemmaModel(configuration)
        case .gemma2:
            let configuration = try JSONDecoder().decode(
-                GemmaConfiguration.self, from: Data(contentsOf: configuration))
+                Gemma2Configuration.self, from: Data(contentsOf: configuration))
            return Gemma2Model(configuration)
        case .qwen2:
            let configuration = try JSONDecoder().decode(