mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 07:57:44 +00:00
Yuante Li
d1019cb568
* fix model input shape bug and costeer_model bug * fix a bug * fix a bug in docker result extraction * a system-level optimization * add a filter of stdout * update * add stdout to model * model training_hyperparameters update * quant scenario * update some quant settings * llm choose action * Thompson Sampling Bandit for action choosing * refine both scens * add trace messages for quant scen * fix some bugs * fix some bugs * update * update * update * fix * fix * fix * update for merge * fix ci * fix some bugs * fix ci * fix ci * fix ci * fix ci * refactor * default qlib4rdagent local env downloading * fix ci * fix ci * fix a bug * fix ci * fix: align all prompts on template (#908) * use template to render all prompts * fix CI --------- Co-authored-by: Xu Yang <xuyang1@microsoft.com> * add fin_quant in cli * fix a bug * fix ci * fix some bugs * refactor * remove the columns in hypothesis if no value generated in this column * fix a bug * fix ci * fix conda env * add qlib gitignore * remove existed qlib folder & install torch in qlib conda * fix workspace ui in feedback * align model config in coder and runner in docker or conda * fix CI * fix CI --------- Co-authored-by: Xu Yang <peteryang@vip.qq.com> Co-authored-by: Xu Yang <xuyang1@microsoft.com>
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()
|