Update models to match actual behavior

- ResScripts
- ResEmbeddings
This commit is contained in:
awsr
2026-03-27 14:05:41 -07:00
parent 6eadf40f33
commit fe7e4b40ff
+5 -5
View File
@@ -484,17 +484,17 @@ else:
FlagsModel = create_model("Flags", __config__=pydantic_config, **flags)
class ResEmbeddings(BaseModel):
loaded: list = Field(default=None, title="loaded", description="List of loaded embeddings")
skipped: list = Field(default=None, title="skipped", description="List of skipped embeddings")
loaded: list = Field(title="loaded", description="List of loaded embeddings")
skipped: list = Field(title="skipped", description="List of skipped embeddings")
class ResMemory(BaseModel):
ram: dict = Field(title="RAM", description="System memory stats")
cuda: dict = Field(title="CUDA", description="nVidia CUDA memory stats")
class ResScripts(BaseModel):
txt2img: list = Field(default=None, title="Txt2img", description="Titles of scripts (txt2img)")
img2img: list = Field(default=None, title="Img2img", description="Titles of scripts (img2img)")
control: list = Field(default=None, title="Control", description="Titles of scripts (control)")
txt2img: list[str] = Field(title="Txt2img", description="Titles of scripts (txt2img)")
img2img: list[str] = Field(title="Img2img", description="Titles of scripts (img2img)")
control: list[str] = Field(title="Control", description="Titles of scripts (control)")
class ResGPU(BaseModel): # definition of http response
name: str = Field(title="GPU Name", description="GPU device name")