Update config.yaml

config.yaml

@@ -1,62 +1,64 @@
-default_variant: "8B"
+# Default model size (can be overridden at runtime)
+default_model_size: "8B"
 
-# ============================================================
-# Variant-specific settings
-# ============================================================
-variants:
-  8B:
+# ==============================================================================
+# Model-specific configurations
+# ==============================================================================
+models:
+  "8B":
     tokens:
       num_mask_and_eom: 152066 
       mask: 152065  
       eom: 152066  
-      num_end_token: 5
-      causal_pad: 3
-      svg_end: 1
       base_offset: 152064
     coordinates:
-      bbox: 200
       pix_pad_offset: 152071
       coord_pad_offset: 152071  
     colors:
-      cmd_fill: 40010
-      color_token_start: 40010
-      max_color_tokens: 4098
       color_start_offset: 192075
       color_end_offset: 196172
-      default_color_token: 40010
-      black_color_token: 40012
+    model:
+      vocab_size: 197000
     huggingface:
       qwen_model: "Qwen/Qwen2.5-VL-7B-Instruct"
       omnisvg_model: "OmniSVG/OmniSVG1.1_8B"
 
-  4B:
+  "4B":
     tokens:
       num_mask_and_eom: 151938 
       mask: 151937 
       eom: 151938 
-      num_end_token: 5
-      causal_pad: 3
-      svg_end: 1
       base_offset: 151936
     coordinates:
-      bbox: 200
       pix_pad_offset: 151943
       coord_pad_offset: 151943  
     colors:
-      cmd_fill: 40010
-      color_token_start: 40010
-      max_color_tokens: 4098
       color_start_offset: 191946
       color_end_offset: 196044
-      default_color_token: 40010
-      black_color_token: 40012
+    model:
+      vocab_size: 197000
     huggingface:
       qwen_model: "Qwen/Qwen2.5-VL-3B-Instruct"
       omnisvg_model: "OmniSVG/OmniSVG1.1_4B"
 
-# ============================================================
-# Common settings (shared by all variants)
-# ============================================================
+# ==============================================================================
+# Shared configurations (same for both models)
+# ==============================================================================
+tokens:
+  num_end_token: 5
+  causal_pad: 3
+  svg_end: 1
+
+coordinates:
+  bbox: 200
+
+colors:
+  cmd_fill: 40010
+  color_token_start: 40010
+  max_color_tokens: 4098
+  default_color_token: 40010
+  black_color_token: 40012  
+
 svg_commands:
   move: -5
   line: -4
@@ -72,7 +74,7 @@ model:
   bos_token_id: 196998
   eos_token_id: 196999
   pad_token_id: 151643
-  max_length: 1024
+  max_length: 1536
 
 text:
   max_length: 200