banhxeo.model.config module

class banhxeo.model.config.ModelConfig(*, vocab_size: int | None = None)[source]

Bases: BaseModel

vocab_size: int | None

model_config: ClassVar[ConfigDict] = {}: Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

class banhxeo.model.config.GenerateConfig(*, sampling: str = 'greedy', max_length: int | None = None, k: int | None = None, p: float | None = None, temp: int | None = None)[source]

Bases: BaseModel

sampling: str

max_length: int | None

k: int | None

p: float | None

temp: int | None

model_config: ClassVar[ConfigDict] = {}: Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].