Unverified Commit 12dc8e09 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #6584 from vladmandic/fix-get-memory

follow-up for pr #6466 to relax response type check enforcement in get_memory
parents 3fe9e9e5 2275f130
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -262,5 +262,5 @@ class EmbeddingsResponse(BaseModel):
    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")
    ram: dict = Field(title="RAM", description="System memory stats")
    cuda: dict = Field(title="CUDA", description="nVidia CUDA memory stats")