mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-27 23:47:46 +00:00
14 lines
446 B
Python
14 lines
446 B
Python
|
from pydantic_settings import SettingsConfigDict
|
||
|
|
|
||
|
|
from rdagent.components.coder.CoSTEER.config import CoSTEERSettings
|
||
|
|
|
||
|
|
|
||
|
|
class ModelCoSTEERSettings(CoSTEERSettings):
|
||
|
|
model_config = SettingsConfigDict(env_prefix="MODEL_CoSTEER_")
|
||
|
|
|
||
|
|
env_type: str = "conda" # or "docker"
|
||
|
|
"""Environment to run model code in coder and runner: 'conda' for local conda env, 'docker' for Docker container"""
|
||
|
|
|
||
|
|
|
||
|
|
MODEL_COSTEER_SETTINGS = ModelCoSTEERSettings()
|