Skip to content

Commit 7b8c796

Browse files
authored
Support more LoRA models in WandbDeploymentConfig (#495)
* Support more LoRA models in WandbDeploymentConfig * Correct Llama names
1 parent f822130 commit 7b8c796

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

src/art/utils/deployment/wandb.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,8 @@ class WandbDeploymentConfig(DeploymentConfig):
1515
"""Configuration for deploying to W&B.
1616
1717
Supported base models:
18+
- meta-llama/Llama-3.1-8B-Instruct
19+
- meta-llama/Llama-3.1-70B-Instruct
1820
- OpenPipe/Qwen3-14B-Instruct
1921
- Qwen/Qwen2.5-14B-Instruct
2022
"""
@@ -23,6 +25,8 @@ class WandbDeploymentConfig(DeploymentConfig):
2325

2426

2527
WANDB_SUPPORTED_BASE_MODELS = [
28+
"meta-llama/Llama-3.1-8B-Instruct",
29+
"meta-llama/Llama-3.1-70B-Instruct",
2630
"OpenPipe/Qwen3-14B-Instruct",
2731
"Qwen/Qwen2.5-14B-Instruct",
2832
]

0 commit comments

Comments
 (0)