KAPO hf runtime finalizer

.env

@@ -8,21 +8,21 @@ BRAIN_ROLES=debugger,fallback
 BRAIN_TEMPLATE=hf-space-cpu
 BRAIN_TUNNEL_PROVIDER=none
 FIREBASE_ENABLED=0
-GOOGLE_DRIVE_BOOTSTRAP_URL=https://drive.google.com/uc?export=download&id=19jyBWsQ9ciJVPi2PUigu5ti3gJ24A6TG
+GOOGLE_DRIVE_BOOTSTRAP_URL=https://drive.google.com/uc?export=download&id=1cTITRABqhuRUyCaHFRA8p3wGNQeU0DK-
 HF_ACCELERATOR=cpu
 HF_SPACE_DOCKER=1
 KAGGLE_AUTO_BOOTSTRAP=0
-KAPO_BOOTSTRAP_URL=https://drive.google.com/uc?export=download&id=19jyBWsQ9ciJVPi2PUigu5ti3gJ24A6TG
+KAPO_BOOTSTRAP_URL=https://drive.google.com/uc?export=download&id=1cTITRABqhuRUyCaHFRA8p3wGNQeU0DK-
 KAPO_COMPUTE_PROFILE=cpu
 KAPO_DISABLE_EMBEDDINGS=1
 KAPO_HF_INFERENCE_API=0
 KAPO_HF_TRANSFORMERS_RUNTIME=1
 KAPO_LAZY_EMBED_STARTUP=1
 KAPO_LAZY_MODEL_STARTUP=1
-KAPO_PATCH_BUNDLE_URL=https://drive.google.com/uc?export=download&id=1dj3gorVxheXtEjDqydavLkN-qtEJz158
-KAPO_PATCH_MANIFEST_URL=https://drive.google.com/uc?export=download&id=1dD1YCMloO4bRruWO4r0QJaMxfIsRDyNb
-KAPO_REMOTE_ENV_PASSWORD_B64=SGhwOW1BdWt1UFNyd1pYTlJPckFHZjA2ZVVJTXZYTmo
-KAPO_REMOTE_ENV_URL_B64=aHR0cHM6Ly9kcml2ZS5nb29nbGUuY29tL3VjP2V4cG9ydD1kb3dubG9hZCZpZD0xdFI4XzdxVTFxNDBnWjBpS1BRWXFPemtGRmlJeWFOQ0U
+KAPO_PATCH_BUNDLE_URL=https://drive.google.com/uc?export=download&id=1be4XGzTO4qm-V3HNuMKZNXD_3Ws-iF9H
+KAPO_PATCH_MANIFEST_URL=https://drive.google.com/uc?export=download&id=1vt8VQ034HjoCzC5SBZcDNK4mclZJzKjv
+KAPO_REMOTE_ENV_PASSWORD_B64=UTBkdEM5R1FxYXRScEp3S3U5dkhMUWtQaC1IY0NjSm8
+KAPO_REMOTE_ENV_URL_B64=aHR0cHM6Ly9kcml2ZS5nb29nbGUuY29tL3VjP2V4cG9ydD1kb3dubG9hZCZpZD0xNUw5UXAzS3E0ZTZ5RVg4VHlFLVRZMks0ZHJnMWhCUEc
 KAPO_SHARED_STATE_BACKEND=google_drive
 MODEL_PROFILE_ID=hf-debugger-qwen25-7b-instruct
 MODEL_REPO=Qwen/Qwen2.5-1.5B-Instruct

bootstrap_space_runtime.py

@@ -16,21 +16,21 @@ DEFAULT_ENV = {
   "BRAIN_TEMPLATE": "hf-space-cpu",
   "BRAIN_TUNNEL_PROVIDER": "none",
   "FIREBASE_ENABLED": "0",
-  "GOOGLE_DRIVE_BOOTSTRAP_URL": "https://drive.google.com/uc?export=download&id=19jyBWsQ9ciJVPi2PUigu5ti3gJ24A6TG",
+  "GOOGLE_DRIVE_BOOTSTRAP_URL": "https://drive.google.com/uc?export=download&id=1cTITRABqhuRUyCaHFRA8p3wGNQeU0DK-",
   "HF_ACCELERATOR": "cpu",
   "HF_SPACE_DOCKER": "1",
   "KAGGLE_AUTO_BOOTSTRAP": "0",
-  "KAPO_BOOTSTRAP_URL": "https://drive.google.com/uc?export=download&id=19jyBWsQ9ciJVPi2PUigu5ti3gJ24A6TG",
+  "KAPO_BOOTSTRAP_URL": "https://drive.google.com/uc?export=download&id=1cTITRABqhuRUyCaHFRA8p3wGNQeU0DK-",
   "KAPO_COMPUTE_PROFILE": "cpu",
   "KAPO_DISABLE_EMBEDDINGS": "1",
   "KAPO_HF_INFERENCE_API": "0",
   "KAPO_HF_TRANSFORMERS_RUNTIME": "1",
   "KAPO_LAZY_EMBED_STARTUP": "1",
   "KAPO_LAZY_MODEL_STARTUP": "1",
-  "KAPO_PATCH_BUNDLE_URL": "https://drive.google.com/uc?export=download&id=1dj3gorVxheXtEjDqydavLkN-qtEJz158",
-  "KAPO_PATCH_MANIFEST_URL": "https://drive.google.com/uc?export=download&id=1dD1YCMloO4bRruWO4r0QJaMxfIsRDyNb",
-  "KAPO_REMOTE_ENV_PASSWORD_B64": "SGhwOW1BdWt1UFNyd1pYTlJPckFHZjA2ZVVJTXZYTmo",
-  "KAPO_REMOTE_ENV_URL_B64": "aHR0cHM6Ly9kcml2ZS5nb29nbGUuY29tL3VjP2V4cG9ydD1kb3dubG9hZCZpZD0xdFI4XzdxVTFxNDBnWjBpS1BRWXFPemtGRmlJeWFOQ0U",
+  "KAPO_PATCH_BUNDLE_URL": "https://drive.google.com/uc?export=download&id=1be4XGzTO4qm-V3HNuMKZNXD_3Ws-iF9H",
+  "KAPO_PATCH_MANIFEST_URL": "https://drive.google.com/uc?export=download&id=1vt8VQ034HjoCzC5SBZcDNK4mclZJzKjv",
+  "KAPO_REMOTE_ENV_PASSWORD_B64": "UTBkdEM5R1FxYXRScEp3S3U5dkhMUWtQaC1IY0NjSm8",
+  "KAPO_REMOTE_ENV_URL_B64": "aHR0cHM6Ly9kcml2ZS5nb29nbGUuY29tL3VjP2V4cG9ydD1kb3dubG9hZCZpZD0xNUw5UXAzS3E0ZTZ5RVg4VHlFLVRZMks0ZHJnMWhCUEc",
   "KAPO_SHARED_STATE_BACKEND": "google_drive",
   "MODEL_PROFILE_ID": "hf-debugger-qwen25-7b-instruct",
   "MODEL_REPO": "Qwen/Qwen2.5-1.5B-Instruct",

brain_server/api/main.py

@@ -287,6 +287,9 @@ def _load_default_model() -> None:
     if _hf_transformers_runtime_enabled():
         ensure_model_loaded(repo_id, filename, hf_token=os.getenv("HF_TOKEN"))
         return
+    if not filename and ("huggingface" in provider or "hf-space" in provider):
+        ensure_model_loaded(repo_id, filename, hf_token=os.getenv("HF_TOKEN"))
+        return
     if not filename and (_feature_enabled("KAPO_HF_INFERENCE_API", default=False) or "huggingface" in provider or "hf-space" in provider):
         MODEL = None
         MODEL_ERROR = None
@@ -2314,7 +2317,7 @@ def ensure_model_loaded(repo_id: str, filename: str, hf_token: str | None = None
         MODEL = None
         MODEL_ERROR = 'model repo missing'
         return
-    if _kapo_hf_transformers_enabled():
+    if _kapo_hf_transformers_enabled() or (_is_hf_space_runtime() and not filename):
         try:
             from transformers import AutoModelForCausalLM, AutoTokenizer
             tokenizer = AutoTokenizer.from_pretrained(repo_id, token=hf_token, trust_remote_code=True)

kapo.env

@@ -8,21 +8,21 @@ BRAIN_ROLES=debugger,fallback
 BRAIN_TEMPLATE=hf-space-cpu
 BRAIN_TUNNEL_PROVIDER=none
 FIREBASE_ENABLED=0
-GOOGLE_DRIVE_BOOTSTRAP_URL=https://drive.google.com/uc?export=download&id=19jyBWsQ9ciJVPi2PUigu5ti3gJ24A6TG
+GOOGLE_DRIVE_BOOTSTRAP_URL=https://drive.google.com/uc?export=download&id=1cTITRABqhuRUyCaHFRA8p3wGNQeU0DK-
 HF_ACCELERATOR=cpu
 HF_SPACE_DOCKER=1
 KAGGLE_AUTO_BOOTSTRAP=0
-KAPO_BOOTSTRAP_URL=https://drive.google.com/uc?export=download&id=19jyBWsQ9ciJVPi2PUigu5ti3gJ24A6TG
+KAPO_BOOTSTRAP_URL=https://drive.google.com/uc?export=download&id=1cTITRABqhuRUyCaHFRA8p3wGNQeU0DK-
 KAPO_COMPUTE_PROFILE=cpu
 KAPO_DISABLE_EMBEDDINGS=1
 KAPO_HF_INFERENCE_API=0
 KAPO_HF_TRANSFORMERS_RUNTIME=1
 KAPO_LAZY_EMBED_STARTUP=1
 KAPO_LAZY_MODEL_STARTUP=1
-KAPO_PATCH_BUNDLE_URL=https://drive.google.com/uc?export=download&id=1dj3gorVxheXtEjDqydavLkN-qtEJz158
-KAPO_PATCH_MANIFEST_URL=https://drive.google.com/uc?export=download&id=1dD1YCMloO4bRruWO4r0QJaMxfIsRDyNb
-KAPO_REMOTE_ENV_PASSWORD_B64=SGhwOW1BdWt1UFNyd1pYTlJPckFHZjA2ZVVJTXZYTmo
-KAPO_REMOTE_ENV_URL_B64=aHR0cHM6Ly9kcml2ZS5nb29nbGUuY29tL3VjP2V4cG9ydD1kb3dubG9hZCZpZD0xdFI4XzdxVTFxNDBnWjBpS1BRWXFPemtGRmlJeWFOQ0U
+KAPO_PATCH_BUNDLE_URL=https://drive.google.com/uc?export=download&id=1be4XGzTO4qm-V3HNuMKZNXD_3Ws-iF9H
+KAPO_PATCH_MANIFEST_URL=https://drive.google.com/uc?export=download&id=1vt8VQ034HjoCzC5SBZcDNK4mclZJzKjv
+KAPO_REMOTE_ENV_PASSWORD_B64=UTBkdEM5R1FxYXRScEp3S3U5dkhMUWtQaC1IY0NjSm8
+KAPO_REMOTE_ENV_URL_B64=aHR0cHM6Ly9kcml2ZS5nb29nbGUuY29tL3VjP2V4cG9ydD1kb3dubG9hZCZpZD0xNUw5UXAzS3E0ZTZ5RVg4VHlFLVRZMks0ZHJnMWhCUEc
 KAPO_SHARED_STATE_BACKEND=google_drive
 MODEL_PROFILE_ID=hf-debugger-qwen25-7b-instruct
 MODEL_REPO=Qwen/Qwen2.5-1.5B-Instruct