Fix model name

yuiseki · Apr 19, 2024 · 84cef51 · 84cef51
1 parent aa186ae
commit 84cef51
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/recipes/A5000_24GB_x8/gemma-2b/gemma-2b-aya-ja.yaml b/recipes/A5000_24GB_x8/gemma-2b/gemma-2b-aya-ja.yaml
@@ -1,6 +1,6 @@
 target_task: tasks/i18n/ja.md
 base_model_id: google/gemma-2b
-model_name: gemma-2b-aya-v0.1
+model_name: gemma-2b-aya-ja-v0.1
 output_base_dir: output
 dataset_id: CohereForAI/aya_dataset
 dataset_input_field_name: inputs

diff --git a/.../A5000_24GB_x8/gemma-7b/gemma-7b-aya.yaml → ...000_24GB_x8/gemma-7b/gemma-7b-aya-ja.yaml b/.../A5000_24GB_x8/gemma-7b/gemma-7b-aya.yaml → ...000_24GB_x8/gemma-7b/gemma-7b-aya-ja.yaml
@@ -1,7 +1,7 @@
 target_task: tasks/i18n/ja.md
 base_model_id: google/gemma-7b
-model_name: gemma-7b-aya-v0.1
-output_base_dir: /data/output
+model_name: gemma-7b-aya-ja-v0.1
+output_base_dir: output
 dataset_id: CohereForAI/aya_dataset
 dataset_input_field_name: inputs
 dataset_output_field_name: targets