Add interface for using generic CoreML LLMs
Compare changes
Files
3- Abhinav Jain authored
@@ -44,7 +44,7 @@ public macro MODEL_MAKE_PREDICTION(_ input_name: Any, _ attention_ids: Any, _ ou
@@ -44,7 +44,7 @@ public macro MODEL_MAKE_PREDICTION(_ input_name: Any, _ attention_ids: Any, _ ou
@@ -72,8 +72,8 @@ class CoreMLEncoder<Scalar: BinaryFloatingPoint & Codable>: SNLPEncoder {
@@ -72,8 +72,8 @@ class CoreMLEncoder<Scalar: BinaryFloatingPoint & Codable>: SNLPEncoder {
@@ -83,24 +83,20 @@ public class MiniLMEmbeddings {
@@ -83,24 +83,20 @@ public class MiniLMEmbeddings {