Skip to content

Commit 2a025fe

Browse files
new apriel ckpt
1 parent 89c7577 commit 2a025fe

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_apriel.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,9 @@
3939
chat_model_args=CHAT_MODEL_ARGS_DICT["apriel/slam-15b"],
4040
flags=apriel_flags,
4141
)
42-
AGENT_SLAM_15B.chat_model_args.base_url = "https://infer-rc10-runai-nowllm.inference.ta121237.dgxcloud.ai/v1"
42+
AGENT_SLAM_15B.chat_model_args.base_url = "https://apriel1p6-15b-runai-nowllm.inference.ta121237.dgxcloud.ai/v1"
4343
AGENT_SLAM_15B.chat_model_args.api_key = "-cPh_74qWJBTooW6MTf6ew:R1zzSE2wSYHwy8CdvPx3s3xdCuQRbhSiY7ydscQEOZw"
44-
AGENT_SLAM_15B.chat_model_args.model_name = "openai/Slam-15B"
44+
AGENT_SLAM_15B.chat_model_args.model_name = "Apriel-1p6-15B-Thinker"
4545

4646
# Run study
4747
study = make_study(

0 commit comments

Comments
 (0)