mirror of
https://github.com/azaion/gps-denied-desktop.git
synced 2026-04-22 22:36:36 +00:00
Oleksandr Bezdieniezhnykh
11 lines
192 B
Python
11 lines
192 B
Python
from pydantic import BaseModel
|
|
|
|
|
|
class ModelConfig(BaseModel):
|
|
model_name: str
|
|
model_path: str
|
|
format: str = "tensorrt"
|
|
precision: str = "fp16"
|
|
warmup_iterations: int = 3
|
|
|