samwaugh commited on
Commit
f79ee4f
·
1 Parent(s): 7f64014

Work with new embeddings

Browse files
Files changed (1) hide show
  1. 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 / "PaintingClip"
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
- # Add below existing EMBEDDINGS_DIR constants
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"