@@ -341,7 +341,7 @@ In [VLMModelFactory.swift](VLMModelFactory.swift) register the model type itself
341
341
(this is independent of the model id):
342
342
343
343
```swift
344
- public class ModelTypeRegistry : @unchecked Sendable {
344
+ public class VLMTypeRegistry : @unchecked Sendable {
345
345
...
346
346
private var creators: [String : @Sendable (URL) throws -> any LanguageModel] = [
347
347
" yourModel" : create (YourModelConfiguration.self , YourModel.init ),
@@ -350,19 +350,19 @@ public class ModelTypeRegistry: @unchecked Sendable {
350
350
Similarly, register the UserInputProcessor type (`preprocessor_config.json `):
351
351
352
352
```swift
353
- public class ProcessorTypeRegistry : @unchecked Sendable {
353
+ public class VLMProcessorTypeRegistry : @unchecked Sendable {
354
354
...
355
355
private var creators:
356
356
[String : @Sendable (URL, any Tokenizer) throws -> any UserInputProcessor] = [
357
357
" YourModelProcessor" : create (
358
358
YourModelProcessorConfiguration.self , YourModelProcessor.init ),
359
359
```
360
360
361
- Add a constant for the model in the ModelRegistry (not strictly required but useful
361
+ Add a constant for the model in the VLMRegistry (not strictly required but useful
362
362
for callers to refer to it in code):
363
363
364
364
```swift
365
- public class ModelRegistry : @unchecked Sendable {
365
+ public class VLMRegistry : @unchecked Sendable {
366
366
...
367
367
static public let yourModel_4bit = ModelConfiguration (
368
368
id : " mlx-community/YourModel-4bit" ,
0 commit comments