Skip to content

Commit

Permalink
Update utils.py
Browse files Browse the repository at this point in the history
  • Loading branch information
SLR722 authored Dec 30, 2024
1 parent 9f542b9 commit 133a354
Showing 1 changed file with 1 addition and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -82,6 +82,7 @@ class DatasetSchema(BaseModel):

def _validate_model_id(model_id: str) -> Model:
model = resolve_model(model_id)
print('print model', model.core_model_id.value)
if model is None or model.core_model_id.value not in MODEL_CONFIGS:
raise ValueError(f"Model {model_id} is not supported.")
return model
Expand Down

0 comments on commit 133a354

Please sign in to comment.