Update recipes for RTX_3060_12GB

yuiseki · yuiseki · commit 934eec176fc4 · 2024-03-26T16:26:58.000+09:00
diff --git a/recipes/RTX_3060_12GB/fake-news-detector.yaml b/recipes/RTX_3060_12GB/fake-news-detector.yaml
@@ -1,5 +1,5 @@
 target_task: tasks/nlp/fake-news-detection.md
-base_model_id: TinyLlama/TinyLlama-1.1B-Chat-v1.0
+base_model_id: TinyLlama/TinyLlama-1.1B-intermediate-step-715k-1.5T
 model_name: tinyllama-fake-news-detector-v1
 output_base_dir: output
 dataset_id: mrm8488/fake-news
diff --git a/recipes/RTX_3060_12GB/sentiment-analyzer.yaml b/recipes/RTX_3060_12GB/sentiment-analyzer.yaml
@@ -1,5 +1,5 @@
 target_task: tasks/nlp/hate-speech-detection.md
-base_model_id: TinyLlama/TinyLlama-1.1B-Chat-v1.0
+base_model_id: TinyLlama/TinyLlama-1.1B-intermediate-step-715k-1.5T
 model_name: tinyllama-sentiment-analyzer-v1
 output_base_dir: output
 dataset_id: carblacac/twitter-sentiment-analysis
diff --git a/recipes/RTX_3060_12GB/sql-coder.yaml b/recipes/RTX_3060_12GB/sql-coder.yaml
@@ -1,5 +1,5 @@
 target_task: tasks/text-generation/text2sql.md
-base_model_id: TinyLlama/TinyLlama-1.1B-Chat-v1.0
+base_model_id: TinyLlama/TinyLlama-1.1B-intermediate-step-715k-1.5T
 model_name: tinyllama-sql-coder-v1
 output_base_dir: output
 dataset_id: b-mc2/sql-create-context
