hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 11:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #6466 from vladmandic/api-get-memory

Browse Source 
Implement API get-memory
This commit is contained in:
AUTOMATIC1111
2023-01-10 02:02:19 +03:00
committed by GitHub
parent 50f91c2945 95727312ca
commit b1d976dca2
2 changed files with 39 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/api/models.py
View File

@@ -260,3 +260,7 @@ class EmbeddingItem(BaseModel):
class EmbeddingsResponse(BaseModel):
loaded: Dict[str, EmbeddingItem] = Field(title="Loaded", description="Embeddings loaded for the current model")
skipped: Dict[str, EmbeddingItem] = Field(title="Skipped", description="Embeddings skipped for the current model (likely due to architecture incompatibility)")
class MemoryResponse(BaseModel):
ram: dict[str, str] | dict[str, float] = Field(title="RAM", description="System memory stats")
cuda: dict[str, str] | dict[str, dict] = Field(title="CUDA", description="nVidia CUDA memory stats")