diff --git a/src/art/utils/deployment/wandb.py b/src/art/utils/deployment/wandb.py index a02ff7dac..94a324691 100644 --- a/src/art/utils/deployment/wandb.py +++ b/src/art/utils/deployment/wandb.py @@ -15,6 +15,8 @@ class WandbDeploymentConfig(DeploymentConfig): """Configuration for deploying to W&B. Supported base models: + - meta-llama/Llama-3.1-8B-Instruct + - meta-llama/Llama-3.1-70B-Instruct - OpenPipe/Qwen3-14B-Instruct - Qwen/Qwen2.5-14B-Instruct """ @@ -23,6 +25,8 @@ class WandbDeploymentConfig(DeploymentConfig): WANDB_SUPPORTED_BASE_MODELS = [ + "meta-llama/Llama-3.1-8B-Instruct", + "meta-llama/Llama-3.1-70B-Instruct", "OpenPipe/Qwen3-14B-Instruct", "Qwen/Qwen2.5-14B-Instruct", ]