Rulga commited on
Commit
8db23d8
·
1 Parent(s): 7aaf8d4

Add model paths and create directories for training output

Browse files
Files changed (1) hide show
  1. config/settings.py +10 -1
config/settings.py CHANGED
@@ -18,6 +18,15 @@ load_dotenv(verbose=True)
18
  BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
19
  VECTOR_STORE_PATH = os.path.join(BASE_DIR, "data", "vector_store")
20
 
 
 
 
 
 
 
 
 
 
21
  # Model settings
22
  EMBEDDING_MODEL = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
23
  DEFAULT_MODEL = "HuggingFaceH4/zephyr-7b-beta"
@@ -28,4 +37,4 @@ if not HF_TOKEN:
28
  raise ValueError("HUGGINGFACE_TOKEN not found in environment variables")
29
 
30
  # Request settings
31
- USER_AGENT = "Status-Law-Assistant/1.0"
 
18
  BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
19
  VECTOR_STORE_PATH = os.path.join(BASE_DIR, "data", "vector_store")
20
 
21
+ # Add missing paths for training models
22
+ MODEL_PATH = os.path.join(BASE_DIR, "models")
23
+ TRAINING_OUTPUT_DIR = os.path.join(BASE_DIR, "models", "trained")
24
+
25
+ # Create directories if they don't exist
26
+ os.makedirs(VECTOR_STORE_PATH, exist_ok=True)
27
+ os.makedirs(MODEL_PATH, exist_ok=True)
28
+ os.makedirs(TRAINING_OUTPUT_DIR, exist_ok=True)
29
+
30
  # Model settings
31
  EMBEDDING_MODEL = "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
32
  DEFAULT_MODEL = "HuggingFaceH4/zephyr-7b-beta"
 
37
  raise ValueError("HUGGINGFACE_TOKEN not found in environment variables")
38
 
39
  # Request settings
40
+ USER_AGENT = "Status-Law-Assistant/1.0"