Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix agent scheduler with pydantic 2

Browse Source
This commit is contained in:
Disty0
2025-09-04 23:38:45 +03:00
parent aec30cc063
commit c210c93444
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/api/models.py
+12
View File
@@ -21,6 +21,14 @@ class ModelDef(BaseModel):
field_exclude: bool = False
class DummyConfig:
dummy_value = None
if not hasattr(BaseModel, "__config__"):
BaseModel.__config__ = DummyConfig
class PydanticModelGenerator:
def __init__(
self,
@@ -210,6 +218,8 @@ ReqTxt2Img = PydanticModelGenerator(
{"key": "extra", "type": Optional[dict], "default": {}, "exclude": True},
]
).generate_model()
if not hasattr(ReqTxt2Img, "__config__"):
ReqTxt2Img.__config__ = DummyConfig
StableDiffusionTxt2ImgProcessingAPI = ReqTxt2Img
class ResTxt2Img(BaseModel):
@@ -238,6 +248,8 @@ ReqImg2Img = PydanticModelGenerator(
{"key": "extra", "type": Optional[dict], "default": {}, "exclude": True},
]
).generate_model()
if not hasattr(ReqImg2Img, "__config__"):
ReqImg2Img.__config__ = DummyConfig
StableDiffusionImg2ImgProcessingAPI = ReqImg2Img
class ResImg2Img(BaseModel):