@@ -139,7 +139,7 @@ class ModelConfigBase(ABC, BaseModel):
139
139
140
140
@staticmethod
141
141
def json_schema_extra (schema : dict [str , Any ]) -> None :
142
- schema ["required" ].extend (["key" , "base" , " type" , "format" ])
142
+ schema ["required" ].extend (["key" , "type" , "format" ])
143
143
144
144
model_config = ConfigDict (validate_assignment = True , json_schema_extra = json_schema_extra )
145
145
@@ -175,7 +175,7 @@ def __init_subclass__(cls, **kwargs):
175
175
if issubclass (cls , LegacyProbeMixin ):
176
176
ModelConfigBase .USING_LEGACY_PROBE .add (cls )
177
177
# Cannot use `elif isinstance(cls, UnknownModelConfig)` because UnknownModelConfig is not defined yet
178
- elif cls . __name__ != "UnknownModelConfig" :
178
+ else :
179
179
ModelConfigBase .USING_CLASSIFY_API .add (cls )
180
180
181
181
@staticmethod
@@ -290,6 +290,12 @@ def matches(cls, mod: ModelOnDisk) -> bool:
290
290
def parse (cls , mod : ModelOnDisk ) -> dict [str , Any ]:
291
291
return {}
292
292
293
+ @staticmethod
294
+ def json_schema_extra (schema : dict [str , Any ]) -> None :
295
+ schema ["required" ] = list (set (schema ["required" ].extend (["key" , "base" , "type" , "format" ])))
296
+
297
+ model_config = ConfigDict (validate_assignment = True , json_schema_extra = json_schema_extra )
298
+
293
299
294
300
class CheckpointConfigBase (ABC , BaseModel ):
295
301
"""Base class for checkpoint-style models."""
0 commit comments