update: export from starry-refactor 2026-02-20 14:58

docker-entrypoint.sh

@@ -153,16 +153,18 @@ export PREDICTOR_BRACKETS="${PREDICTOR_BRACKETS:-tcp://127.0.0.1:12028}"
 
         state_yaml="$MODELS_BASE/starry-dist/$model_path/.state.yaml"
         if [ -f "$state_yaml" ]; then
-          model_file=$(sed -n 's/^file: *//p' "$state_yaml" | head -1)
+          # Model weight file is in "best:" field (not "file:")
+          model_file=$(sed -n 's/^best: *//p' "$state_yaml" | head -1)
           if [ -n "$model_file" ]; then
             hf_download "starry-dist" "$model_path/$model_file"
           fi
-          sed -n '/^subs:/,/^[a-z]/{ /^  - /p }' "$state_yaml" | sed 's/^ *- *//' | while read -r sub; do
+          # Sub-models listed under "subs:" (skip commented lines starting with #)
+          sed -n '/^subs:/,/^[a-z]/{ /^[^#]*- /p }' "$state_yaml" | sed 's/^ *- *//' | while read -r sub; do
             if [ -n "$sub" ]; then
               hf_download "starry-dist" "$model_path/$sub/.state.yaml"
               sub_state="$MODELS_BASE/starry-dist/$model_path/$sub/.state.yaml"
               if [ -f "$sub_state" ]; then
-                sub_file=$(sed -n 's/^file: *//p' "$sub_state" | head -1)
+                sub_file=$(sed -n 's/^best: *//p' "$sub_state" | head -1)
                 if [ -n "$sub_file" ]; then
                   hf_download "starry-dist" "$model_path/$sub/$sub_file"
                 fi
@@ -173,30 +175,27 @@ export PREDICTOR_BRACKETS="${PREDICTOR_BRACKETS:-tcp://127.0.0.1:12028}"
       fi
     done
 
-    # Download OCR/TF models (ocr-dist)
-    for config_var in LOC_MODEL_PATH OCR_CONFIG BRACKETS_CONFIG; do
+    # Download LOC model (a single file, not a directory)
+    if [ -n "$LOC_MODEL_PATH" ]; then
+      hf_download "ocr-dist" "$LOC_MODEL_PATH"
+    fi
+
+    # Download OCR/TF config yamls and all referenced files
+    for config_var in OCR_CONFIG BRACKETS_CONFIG; do
       eval config_path=\$$config_var
       if [ -n "$config_path" ]; then
-        if echo "$config_path" | grep -q '\.yaml$\|\.yml$'; then
-          hf_download "ocr-dist" "$config_path"
-          config_file="$MODELS_BASE/ocr-dist/$config_path"
-          if [ -f "$config_file" ]; then
-            sed -n 's/^ *model_path: *//p' "$config_file" | while read -r mpath; do
-              if [ -n "$mpath" ]; then
-                _dir=$(dirname "$config_path")
-                hf_download "ocr-dist" "$_dir/$mpath"
-              fi
-            done
-          fi
-        else
-          hf_download "ocr-dist" "$config_path/.state.yaml"
-          state_yaml="$MODELS_BASE/ocr-dist/$config_path/.state.yaml"
-          if [ -f "$state_yaml" ]; then
-            model_file=$(sed -n 's/^file: *//p' "$state_yaml" | head -1)
-            if [ -n "$model_file" ]; then
-              hf_download "ocr-dist" "$config_path/$model_file"
+        hf_download "ocr-dist" "$config_path"
+        config_file="$MODELS_BASE/ocr-dist/$config_path"
+        if [ -f "$config_file" ]; then
+          _dir=$(dirname "$config_path")
+          # Download all *_path values referenced in the yaml
+          grep '_path:' "$config_file" | sed 's/^[^:]*: *//' | while read -r ref_path; do
+            if [ -n "$ref_path" ]; then
+              # Resolve ./ relative paths
+              ref_path=$(echo "$ref_path" | sed 's|^\./||')
+              hf_download "ocr-dist" "${_dir:+$_dir/}$ref_path"
             fi
-          fi
+          done
         fi
       fi
     done