Update app.py
Browse files
app.py
CHANGED
@@ -7,8 +7,8 @@ from huggingface_hub import hf_hub_download
|
|
7 |
|
8 |
llm = Llama(
|
9 |
model_path=hf_hub_download(
|
10 |
-
repo_id=os.environ.get("REPO_ID", "
|
11 |
-
filename=os.environ.get("MODEL_FILE", "
|
12 |
),
|
13 |
n_ctx=2048,
|
14 |
n_gpu_layers=50, # change n_gpu_layers if you have more or less VRAM
|
|
|
7 |
|
8 |
llm = Llama(
|
9 |
model_path=hf_hub_download(
|
10 |
+
repo_id=os.environ.get("REPO_ID", "wop/kosmox-gguf"),
|
11 |
+
filename=os.environ.get("MODEL_FILE", "kosmox-gguf-unsloth.Q8_0.gguf"),
|
12 |
),
|
13 |
n_ctx=2048,
|
14 |
n_gpu_layers=50, # change n_gpu_layers if you have more or less VRAM
|