Spaces:
Running
Running
Dan Flower
commited on
Commit
·
cf901fd
1
Parent(s):
1295432
troubleshooting model path issue
Browse files- model/download_model.py +2 -1
- 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", "
|
|
|
|
| 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/
|
| 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"
|