mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 07:57:44 +00:00
720994c8e0
* store code into FBImplementation * fix path related bugs * fix a bug * fix factor related small bugs * re-submit all model related code * new code to model coder * finish the model evolving code --------- Co-authored-by: xuyang1 <xuyang1@microsoft.com>
33 lines
949 B
Python
33 lines
949 B
Python
from pathlib import Path
|
|
from typing import Union
|
|
|
|
from pydantic_settings import BaseSettings
|
|
|
|
|
|
class ModelImplSettings(BaseSettings):
|
|
class Config:
|
|
env_prefix = "MODEL_IMPL_" # Use MODEL_IMPL_ as prefix for environment variables
|
|
|
|
file_based_execution_workspace: str = str(
|
|
(Path().cwd() / "git_ignore_folder" / "model_implementation_workspace").absolute(),
|
|
)
|
|
implementation_execution_cache_location: str = str(
|
|
(Path().cwd() / "git_ignore_folder" / "model_implementation_execution_cache").absolute(),
|
|
)
|
|
|
|
knowledge_base_path: Union[str, None] = None
|
|
new_knowledge_base_path: Union[str, None] = None
|
|
|
|
max_loop: int = 10
|
|
|
|
query_former_trace_limit: int = 5
|
|
query_similar_success_limit: int = 5
|
|
fail_task_trial_limit: int = 20
|
|
|
|
evo_multi_proc_n: int = 1
|
|
|
|
enable_execution_cache: bool = True # whether to enable the execution cache
|
|
|
|
|
|
MODEL_IMPL_SETTINGS = ModelImplSettings()
|