relax reponse type check enforcement

This commit is contained in:
Vladimir Mandic 2023-01-09 21:23:58 -05:00 committed by GitHub
parent 3fe9e9e54d
commit 2275f130bf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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")