Refactor field validation in models.py

tabs/images/models.py

@@ -39,22 +39,22 @@ class BaseReq(BaseModel):
     class Config:
         arbitrary_types_allowed = True
 
-    @field_validator('model', 'negative_prompt', 'embeddings', 'clip_skip', 'controlnet_config', mode='before')
-    def check_model(cls, values):
-        for m in appConfig.IMAGES_MODELS:
-            if isinstance(m, dict) and m.get('repo_id') == values.get('model'):
-                loader = m.get('loader')
-
-                if loader == "flux" and values.get('negative_prompt'):
-                    raise ValueError("Negative prompt is not supported for Flux models.")
-                if loader == "flux" and values.get('embeddings'):
-                    raise ValueError("Embeddings are not supported for Flux models.")
-                if loader == "flux" and values.get('clip_skip'):
-                    raise ValueError("Clip skip is not supported for Flux models.")
-                if loader == "flux" and values.get('controlnet_config'):
-                    if "scribble" in values.get('controlnet_config').controlnets:
-                        raise ValueError("Scribble is not supported for Flux models.")
-        return values
+    # @field_validator('model', 'negative_prompt', 'embeddings', 'clip_skip', 'controlnet_config', mode='before')
+    # def check_model(cls, values):
+    #     for m in appConfig.IMAGES_MODELS:
+    #         if isinstance(m, dict) and m.get('repo_id') == values.get('model'):
+    #             loader = m.get('loader')
+
+    #             if loader == "flux" and values.get('negative_prompt'):
+    #                 raise ValueError("Negative prompt is not supported for Flux models.")
+    #             if loader == "flux" and values.get('embeddings'):
+    #                 raise ValueError("Embeddings are not supported for Flux models.")
+    #             if loader == "flux" and values.get('clip_skip'):
+    #                 raise ValueError("Clip skip is not supported for Flux models.")
+    #             if loader == "flux" and values.get('controlnet_config'):
+    #                 if "scribble" in values.get('controlnet_config').controlnets:
+    #                     raise ValueError("Scribble is not supported for Flux models.")
+    #     return values
 
 
 class BaseImg2ImgReq(BaseReq):