Work with new embeddings
Browse files- backend/runner/config.py +3 -8
backend/runner/config.py
CHANGED
|
@@ -32,7 +32,6 @@ if not os.access(WRITE_ROOT, os.W_OK):
|
|
| 32 |
raise RuntimeError(f"Directory {WRITE_ROOT} is not writable")
|
| 33 |
|
| 34 |
print(f"✅ Using WRITE_ROOT: {WRITE_ROOT}")
|
| 35 |
-
|
| 36 |
print(f"✅ Using READ_ROOT: {DATA_READ_ROOT}")
|
| 37 |
|
| 38 |
# Read-only directories (from repo)
|
|
@@ -41,12 +40,8 @@ JSON_INFO_DIR = DATA_READ_ROOT / "json_info"
|
|
| 41 |
MODELS_DIR = DATA_READ_ROOT / "models"
|
| 42 |
MARKER_DIR = DATA_READ_ROOT / "marker_output"
|
| 43 |
|
| 44 |
-
# Model-specific embedding directories
|
| 45 |
-
CLIP_EMBEDDINGS_DIR = EMBEDDINGS_DIR / "CLIP_Embeddings"
|
| 46 |
-
PAINTINGCLIP_EMBEDDINGS_DIR = EMBEDDINGS_DIR / "PaintingCLIP_Embeddings"
|
| 47 |
-
|
| 48 |
# Model directories
|
| 49 |
-
PAINTINGCLIP_MODEL_DIR = MODELS_DIR / "
|
| 50 |
|
| 51 |
# Writable directories (outside repo)
|
| 52 |
OUTPUTS_DIR = WRITE_ROOT / "outputs"
|
|
@@ -67,9 +62,9 @@ TOPICS_JSON = JSON_INFO_DIR / "topics.json"
|
|
| 67 |
CREATORS_JSON = JSON_INFO_DIR / "creators.json"
|
| 68 |
TOPIC_NAMES_JSON = JSON_INFO_DIR / "topic_names.json"
|
| 69 |
|
| 70 |
-
#
|
| 71 |
CLIP_EMBEDDINGS_ST = EMBEDDINGS_DIR / "clip_embeddings.safetensors"
|
| 72 |
CLIP_SENTENCE_IDS = EMBEDDINGS_DIR / "clip_embeddings_sentence_ids.json"
|
| 73 |
|
| 74 |
PAINTINGCLIP_EMBEDDINGS_ST = EMBEDDINGS_DIR / "paintingclip_embeddings.safetensors"
|
| 75 |
-
PAINTINGCLIP_SENTENCE_IDS = EMBEDDINGS_DIR / "paintingclip_embeddings_sentence_ids.json"
|
|
|
|
| 32 |
raise RuntimeError(f"Directory {WRITE_ROOT} is not writable")
|
| 33 |
|
| 34 |
print(f"✅ Using WRITE_ROOT: {WRITE_ROOT}")
|
|
|
|
| 35 |
print(f"✅ Using READ_ROOT: {DATA_READ_ROOT}")
|
| 36 |
|
| 37 |
# Read-only directories (from repo)
|
|
|
|
| 40 |
MODELS_DIR = DATA_READ_ROOT / "models"
|
| 41 |
MARKER_DIR = DATA_READ_ROOT / "marker_output"
|
| 42 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
# Model directories
|
| 44 |
+
PAINTINGCLIP_MODEL_DIR = MODELS_DIR / "PaintingCLIP"
|
| 45 |
|
| 46 |
# Writable directories (outside repo)
|
| 47 |
OUTPUTS_DIR = WRITE_ROOT / "outputs"
|
|
|
|
| 62 |
CREATORS_JSON = JSON_INFO_DIR / "creators.json"
|
| 63 |
TOPIC_NAMES_JSON = JSON_INFO_DIR / "topic_names.json"
|
| 64 |
|
| 65 |
+
# Embedding files (lowercase for backend compatibility)
|
| 66 |
CLIP_EMBEDDINGS_ST = EMBEDDINGS_DIR / "clip_embeddings.safetensors"
|
| 67 |
CLIP_SENTENCE_IDS = EMBEDDINGS_DIR / "clip_embeddings_sentence_ids.json"
|
| 68 |
|
| 69 |
PAINTINGCLIP_EMBEDDINGS_ST = EMBEDDINGS_DIR / "paintingclip_embeddings.safetensors"
|
| 70 |
+
PAINTINGCLIP_SENTENCE_IDS = EMBEDDINGS_DIR / "paintingclip_embeddings_sentence_ids.json"
|