Dan Flower commited on
Commit
cf901fd
·
1 Parent(s): 1295432

troubleshooting model path issue

Browse files
Files changed (2) hide show
  1. model/download_model.py +2 -1
  2. streamlit_app.py +1 -1
model/download_model.py CHANGED
@@ -3,7 +3,8 @@ from huggingface_hub import hf_hub_download
3
 
4
  def main():
5
  MODEL_REPO = os.getenv("MODEL_REPO", "TheBloke/TinyLlama-1.1B-Chat-v1.0-GGUF")
6
- MODEL_FILE = os.getenv("MODEL_FILE", "TinyLlama-1.1B-Chat-v1.0.Q4_K_M.gguf")
 
7
  MODEL_DIR = pathlib.Path(os.getenv("MODEL_DIR", "/tmp/models"))
8
  MODEL_PATH = MODEL_DIR / MODEL_FILE
9
 
 
3
 
4
  def main():
5
  MODEL_REPO = os.getenv("MODEL_REPO", "TheBloke/TinyLlama-1.1B-Chat-v1.0-GGUF")
6
+ MODEL_FILE = os.getenv("MODEL_FILE", "tinyllama-1.1b-chat-v1.0.Q4_K_M.gguf")
7
+ #MODEL_FILE = os.getenv("MODEL_FILE", "TinyLlama-1.1B-Chat-v1.0.Q4_K_M.gguf")
8
  MODEL_DIR = pathlib.Path(os.getenv("MODEL_DIR", "/tmp/models"))
9
  MODEL_PATH = MODEL_DIR / MODEL_FILE
10
 
streamlit_app.py CHANGED
@@ -8,7 +8,7 @@ sys.path.append(os.path.dirname(os.path.abspath(__file__)))
8
  # Ensure /app is in sys.path so we can import utils.* from anywhere
9
  sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
10
  # Environment setup
11
- os.environ["MODEL_PATH"] = "/tmp/models/TinyLlama-1.1B-Chat-v1.0.Q4_K_M.gguf"
12
  os.environ["STREAMLIT_HOME"] = "/tmp/.streamlit"
13
  os.environ["XDG_CONFIG_HOME"] = "/tmp/.streamlit"
14
  os.environ["BROWSER_GATHER_USAGE_STATS"] = "false"
 
8
  # Ensure /app is in sys.path so we can import utils.* from anywhere
9
  sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
10
  # Environment setup
11
+ os.environ["MODEL_PATH"] = "/tmp/models/tinyllama-1.1b-chat-v1.0.Q4_K_M.gguf"
12
  os.environ["STREAMLIT_HOME"] = "/tmp/.streamlit"
13
  os.environ["XDG_CONFIG_HOME"] = "/tmp/.streamlit"
14
  os.environ["BROWSER_GATHER_USAGE_STATS"] = "false"