move model to bos

DesmonDay · Feb 11, 2025 · 7ec8bdc · 7ec8bdc
1 parent c0042fb
commit 7ec8bdc
Show file tree

Hide file tree

Showing 8 changed files with 1 addition and 454,599 deletions.
diff --git a/tests/trainer/test_moe_unified_checkpoint.py b/tests/trainer/test_moe_unified_checkpoint.py
@@ -38,7 +38,7 @@
 }
 
 moe_arguments = {
-    "model_name_or_path": "./tests/trainer/unified-ckpt-qwen2moe",
+    "model_name_or_path": "__internal_testing__/unified-ckpt-qwen2moe",
     "dataset_name_or_path": "./unified_checkpoint/peft_input/data/",
     "output_dir": "./unified_checkpoint/checkpoints/qwen2moe_sft_ckpts",
     "per_device_train_batch_size": 1,

diff --git a/tests/trainer/unified-ckpt-qwen2moe/config.json b/tests/trainer/unified-ckpt-qwen2moe/config.json
diff --git a/tests/trainer/unified-ckpt-qwen2moe/generation_config.json b/tests/trainer/unified-ckpt-qwen2moe/generation_config.json