Sync from GitHub via hub-sync

README.md

@@ -63,6 +63,8 @@ SMOLNALYSIS_MINICPM_CKAN_RETRIEVAL_TEMPERATURE=0
 SMOLNALYSIS_MINICPM_MAX_NEW_TOKENS=384
 ```
 
+`SMOLNALYSIS_MINICPM_CKAN_RETRIEVAL_ADAPTER_REPO_ID` defaults to the repo above, so the Space uses the uploaded CKAN adapter automatically unless this value is overridden.
+
 Upload the trained adapter from a machine with `HF_TOKEN` set:
 
 ```bash

app/backend/minicpm_transformers.py

@@ -27,6 +27,10 @@ DEFAULT_MODEL_ID = os.getenv("SMOLNALYSIS_MINICPM_TRANSFORMERS_MODEL_ID", os.get
 DEFAULT_MAX_NEW_TOKENS = int(os.getenv("SMOLNALYSIS_MINICPM_MAX_NEW_TOKENS", os.getenv("MAX_TOKENS", "384")))
 DEFAULT_TEMPERATURE = float(os.getenv("SMOLNALYSIS_MINICPM_TEMPERATURE", os.getenv("TEMPERATURE", "0.7")))
 DEFAULT_TOP_P = float(os.getenv("SMOLNALYSIS_MINICPM_TOP_P", os.getenv("TOP_P", "0.95")))
+DEFAULT_CKAN_RETRIEVAL_ADAPTER_REPO_ID = os.getenv(
+    "SMOLNALYSIS_DEFAULT_CKAN_RETRIEVAL_ADAPTER_REPO_ID",
+    "build-small-hackathon/smolnalysis-ckan-retrieval-minicpm5-lora",
+)
 ZERO_GPU_DURATION_SECONDS = int(os.getenv("SMOLNALYSIS_MINICPM_ZEROGPU_DURATION_SECONDS", "120"))
 EAGER_LOAD = os.getenv("SMOLNALYSIS_MINICPM_TRANSFORMERS_EAGER_LOAD", os.getenv("SPACE_ID", "")).casefold() not in {
     "",
@@ -133,7 +137,11 @@ def role_config(role: str) -> TransformersRoleConfig:
 
     default_temperature = "0" if role == "ckan_retrieval" else str(DEFAULT_TEMPERATURE)
     adapter_path = _clean_env_value(_role_env(role, "ADAPTER_PATH"), _clean_env_value(_role_env(role, "LORA_PATH"), ""))
-    adapter_repo_id = _clean_env_value(_role_env(role, "ADAPTER_REPO_ID"), _clean_env_value(_role_env(role, "LORA_REPO_ID"), ""))
+    default_adapter_repo_id = DEFAULT_CKAN_RETRIEVAL_ADAPTER_REPO_ID if role == "ckan_retrieval" else ""
+    adapter_repo_id = _clean_env_value(
+        _role_env(role, "ADAPTER_REPO_ID"),
+        _clean_env_value(_role_env(role, "LORA_REPO_ID"), default_adapter_repo_id),
+    )
     max_new_tokens = int(_clean_env_value(_role_env(role, "MAX_NEW_TOKENS"), str(DEFAULT_MAX_NEW_TOKENS)))
     temperature = float(_clean_env_value(_role_env(role, "TEMPERATURE"), default_temperature))
     top_p = float(_clean_env_value(_role_env(role, "TOP_P"), str(DEFAULT_TOP_P)))