vilarin commited on
Commit
84db5e8
1 Parent(s): c9ba4ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,6 +9,7 @@ from huggingface_hub import hf_hub_download
9
 
10
  HF_TOKEN = os.environ.get("HF_TOKEN", None)
11
  MODEL_ID = "google/gemma-2-27b-it"
 
12
  MODEL_NAME = MODEL_ID.split("/")[-1]
13
  MODEL_FILE = "gemma-2-27b-it-Q4_K_M.gguf"
14
 
@@ -16,7 +17,7 @@ os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
16
 
17
  llm = Llama(
18
  model_path=hf_hub_download(
19
- repo_id=os.environ.get(MODEL_ID),
20
  filename=os.environ.get(MODEL_FILE),
21
  ),
22
  n_ctx=4096,
 
9
 
10
  HF_TOKEN = os.environ.get("HF_TOKEN", None)
11
  MODEL_ID = "google/gemma-2-27b-it"
12
+ REPO_ID = "bartowski/gemma-2-27b-it-GGUF"
13
  MODEL_NAME = MODEL_ID.split("/")[-1]
14
  MODEL_FILE = "gemma-2-27b-it-Q4_K_M.gguf"
15
 
 
17
 
18
  llm = Llama(
19
  model_path=hf_hub_download(
20
+ repo_id=os.environ.get(REPO_ID),
21
  filename=os.environ.get(MODEL_FILE),
22
  ),
23
  n_ctx=4096,