mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 16:07:46 +00:00
6f3a44c18a
* init commit * remove the 5-fold spec from prompts * refine the hyperparameter specification * do not sample data * a small spelling issue * refine prompt to avoid submission cheating * do not sample data * simplify code * refine the coder evaluator prompt * refine wording * remove runtime from proposal * refine wording * refine prompt * add gpu info in runtime_info.py * modify the spec * add router and add refinement exp gen * fix prompt bug * use rule-based logic for router * complete the prompt * fix circular import bug * fix bug * make refine_decision optional * update pipeline prompts: (1) add scenary: in an iterative cooding loop and use sample datasets (2)add some generation tops in coding (3)add evaluation guidelines in evaluation (4)polish the json schema and description * fix a small bug * fix a small bug * rdagent/scenarios/data_science/loop.py back to the original version * refactor: replace _get_exp_gen with default_exp_gen for exp generation * import * refactor: make the __init__ back to main * fix small bugs * fix bugs for proposal_version * move refine into runner * check early stop * EDA improvement & coder classes number * fix CI * slightly refine the prompt * remove rule_base_eval and remove useless prompt --------- Co-authored-by: Xu <v-xuminrui@microsoft.com> Co-authored-by: TPLin22 <tplin2@163.com> Co-authored-by: amstrongzyf <amstrongzyf@126.com> Co-authored-by: Xu Yang <peteryang@vip.qq.com> Co-authored-by: Xu Yang <xuyang1@microsoft.com> Co-authored-by: Young <afe.young@gmail.com>
48 lines
962 B
Plaintext
48 lines
962 B
Plaintext
# Global configs:
|
|
|
|
MAX_RETRY=12000
|
|
RETRY_WAIT_SECONDS=5
|
|
TIMEOUT_FAIL_LIMIT=100
|
|
|
|
# litellm
|
|
# CHAT_MODEL=gpt-4o
|
|
# CHAT_TEMPERATURE=0.7
|
|
|
|
CHAT_STREAM=False
|
|
CHAT_TEMPERATURE=1
|
|
CHAT_MODEL=o1-preview
|
|
SYSTEM_PROMPT_ROLE=user
|
|
|
|
BACKEND=rdagent.oai.backend.LiteLLMAPIBackend
|
|
OPENAI_API_KEY=sk-1234
|
|
OPENAI_API_BASE=http://ep14.213428.xyz:38881
|
|
|
|
|
|
# amc chat model configs:
|
|
EMBEDDING_MODEL=text-embedding-ada-002
|
|
|
|
# Cache Setting (Optional):
|
|
DUMP_CHAT_CACHE=True
|
|
USE_CHAT_CACHE=False
|
|
DUMP_EMBEDDING_CACHE=True
|
|
USE_EMBEDDING_CACHE=False
|
|
LOG_LLM_CHAT_CONTENT=True
|
|
|
|
DS_LOCAL_DATA_PATH=/home/shared/RD-Agent/kaggle
|
|
|
|
DS_IF_USING_MLE_DATA=True
|
|
|
|
|
|
PICKLE_CACHE_FOLDER_PATH_STR=./log/pickle_cache
|
|
CACHE_WITH_PICKLE=False
|
|
ENABLE_CACHE=False
|
|
PROMPT_CACHE_PATH=./log/prompt_cache.db
|
|
|
|
DS_CODER_COSTEER_ENV_TYPE=conda
|
|
# DS_PROPOSAL_VERSION=v2 deprecated
|
|
|
|
DS_CODER_ON_WHOLE_PIPELINE=True
|
|
COSTEER_V2_QUERY_FORMER_TRACE_LIMIT=3
|
|
|
|
# export PYTHONPATH=. # this is for running researcher branch;
|