Skip to content

Commit 663da5d

Browse files
committed
feat(app): expand GPU model and schema for richer hardware specs
Updates SQLModel + Pydantic shape for DiscreteGPU to expose the additional fields populated by the rebuilt GPU dataset. Refs #1
1 parent 399229c commit 663da5d

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

app/models/gpu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ class DiscreteGPU(SQLModel, table=True):
2828
tensor_cores: int | None = None
2929

3030
# Memory
31-
memory_gb: int
31+
memory_gb: float
3232
memory_type: str
3333
memory_bus_bit: int
3434
memory_bandwidth_gbps: float | None = None

app/schemas/gpu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ class GPURead(BaseModel):
2727
stream_processors: int | None = None
2828
rt_cores: int | None = None
2929
tensor_cores: int | None = None
30-
memory_gb: int
30+
memory_gb: float
3131
memory_type: str
3232
memory_bus_bit: int
3333
memory_bandwidth_gbps: float | None = None

0 commit comments

Comments
 (0)