SherinMohamed commited on
Commit
773fbda
·
verified ·
1 Parent(s): 84dd67f

Update app/config.py

Browse files
Files changed (1) hide show
  1. app/config.py +26 -0
app/config.py CHANGED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from dotenv import load_dotenv
3
+
4
+ load_dotenv()
5
+
6
+ class Settings:
7
+ # JAIS
8
+ JAIS_MODEL_ID = os.getenv("JAIS_MODEL_ID", "inceptionai/jais-13b-chat")
9
+ JAIS_MAX_NEW_TOKENS = int(os.getenv("JAIS_MAX_NEW_TOKENS", "220"))
10
+ JAIS_TEMPERATURE = float(os.getenv("JAIS_TEMPERATURE", "0.7"))
11
+ JAIS_TOP_P = float(os.getenv("JAIS_TOP_P", "0.9"))
12
+
13
+ # Whisper
14
+ WHISPER_MODEL_SIZE = os.getenv("WHISPER_MODEL_SIZE", "medium")
15
+
16
+ # XTTS / EGTTS
17
+ XTTS_CONFIG_PATH = os.getenv("XTTS_CONFIG_PATH", "")
18
+ XTTS_VOCAB_PATH = os.getenv("XTTS_VOCAB_PATH", "")
19
+ XTTS_MODEL_DIR = os.getenv("XTTS_MODEL_DIR", "")
20
+ XTTS_SPEAKER_WAV = os.getenv("XTTS_SPEAKER_WAV", "")
21
+
22
+ # Server
23
+ HOST = os.getenv("HOST", "0.0.0.0")
24
+ PORT = int(os.getenv("PORT", "8000"))
25
+
26
+ settings = Settings()