From e7508a12dc9fbba8304d44b5327b12b738af43d0 Mon Sep 17 00:00:00 2001 From: arcticfly Date: Sun, 21 Dec 2025 23:31:34 -0800 Subject: [PATCH 1/2] Support more LoRA models in WandbDeploymentConfig --- src/art/utils/deployment/wandb.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/art/utils/deployment/wandb.py b/src/art/utils/deployment/wandb.py index a02ff7dac..d43a5782e 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", ] From 277cd557b05678deb423acffa05e4f888938ee0b Mon Sep 17 00:00:00 2001 From: arcticfly Date: Sun, 21 Dec 2025 23:55:21 -0800 Subject: [PATCH 2/2] Correct Llama names --- src/art/utils/deployment/wandb.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/art/utils/deployment/wandb.py b/src/art/utils/deployment/wandb.py index d43a5782e..94a324691 100644 --- a/src/art/utils/deployment/wandb.py +++ b/src/art/utils/deployment/wandb.py @@ -15,8 +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 + - meta-llama/Llama-3.1-8B-Instruct + - meta-llama/Llama-3.1-70B-Instruct - OpenPipe/Qwen3-14B-Instruct - Qwen/Qwen2.5-14B-Instruct """ @@ -25,8 +25,8 @@ class WandbDeploymentConfig(DeploymentConfig): WANDB_SUPPORTED_BASE_MODELS = [ - "meta-llama/llama-3.1-8b-instruct", - "meta-llama/llama-3.1-70b-instruct", + "meta-llama/Llama-3.1-8B-Instruct", + "meta-llama/Llama-3.1-70B-Instruct", "OpenPipe/Qwen3-14B-Instruct", "Qwen/Qwen2.5-14B-Instruct", ]