[WIP] Update

models/v0.2.0/project.yml

@@ -39,8 +39,8 @@ description: |
   This step downloads all assets and prepares all the datasets and binaries for
   training use. For example, if you want to train `tl_calamancy_md`, run the following comand:
 
-  ```
-  bash scripts/tl_calamancy_md.sh
+  ```sh
+  MODEL=tl_calamancy_md scripts/train.sh
   ```
   
 
@@ -120,7 +120,7 @@ workflows:
   tl-calamancy-trf:
     - "train-parser-trf"
     - "train-ner-trf"
-    - "assemble"
+    - "assemble-trf"
 
 commands:
   - name: "setup-finetuning-data"

models/v0.2.0/scripts/train.sh

@@ -0,0 +1,26 @@
+#!/bin/bash
+
+MODEL="tl_calamancy_md"  # Set this to tl_calamancy_md, tl_calamancy_lg, or tl_calamancy_trf as needed.
+
+case $MODEL in
+  tl_calamancy_md)
+    echo "Executing actions for tl_calamancy_md..."
+    python -m spacy project run tl-calamancy . \
+        --vars.size md \
+        --vars.vectors vectors/floret-tl-md
+    ;;
+  tl_calamancy_lg)
+    echo "Executing actions for tl_calamancy_lg..."
+    python -m spacy project run tl-calamancy . \
+        --vars.size lg \
+        --vars.vectors vectors/fasttext-tl
+    ;;
+  tl_calamancy_trf)
+    echo "Executing actions for tl_calamancy_trf..."
+    python -m spacy project run tl-calamancy-trf
+    # Add your commands here
+    ;;
+  *)
+    echo "Unknown MODEL: $MODEL"
+    ;;
+esac