2023-11-28 23:50:50 -08:00
|
|
|
from typing import Optional
|
2022-04-27 12:14:58 -07:00
|
|
|
|
2024-01-06 18:32:21 -08:00
|
|
|
from pydantic import BaseModel, ConfigDict
|
2022-04-27 12:14:58 -07:00
|
|
|
|
|
|
|
|
|
2023-11-29 00:39:14 -08:00
|
|
|
|
2023-11-28 23:50:50 -08:00
|
|
|
class JetstreamConfig(BaseModel):
|
|
|
|
|
"""Jetstream configuration."""
|
2023-11-27 16:01:20 -08:00
|
|
|
|
2023-11-28 23:50:50 -08:00
|
|
|
domain: Optional[str] = None
|
2022-04-27 12:14:58 -07:00
|
|
|
|
2023-11-28 23:50:50 -08:00
|
|
|
max_memory: Optional[int] = None
|
2023-05-04 00:58:06 -07:00
|
|
|
|
2023-11-28 23:50:50 -08:00
|
|
|
max_storage: Optional[int] = None
|
2022-04-27 12:14:58 -07:00
|
|
|
|
2023-11-28 23:50:50 -08:00
|
|
|
store_dir: Optional[str] = None
|
2024-01-06 18:32:21 -08:00
|
|
|
|
|
|
|
|
model_config = ConfigDict(protected_namespaces=())
|
