From c660a01bfb77db7737386a0ba31e31ed720c082e Mon Sep 17 00:00:00 2001 From: Yuta Hayashibe Date: Thu, 29 Sep 2022 20:26:03 +0900 Subject: [PATCH] Check model name (openai/whisper@2d3032d) --- whispering/schema.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/whispering/schema.py b/whispering/schema.py index db5b12f..06647e8 100644 --- a/whispering/schema.py +++ b/whispering/schema.py @@ -3,7 +3,7 @@ from typing import List, Optional import torch -from pydantic import BaseModel +from pydantic import BaseModel, root_validator class WhisperConfig(BaseModel): @@ -24,6 +24,15 @@ class WhisperConfig(BaseModel): compression_ratio_threshold: Optional[float] = 2.4 buffer_threshold: Optional[float] = 0.5 + @root_validator + def validate_model_name(cls, values): + if values["model_name"].endswith(".en") and values["language"] not in { + "en", + "English", + }: + raise ValueError("English only model") + return values + class Context(BaseModel, arbitrary_types_allowed=True): timestamp: float = 0.0