lilblueyes commited on
Commit
bb615f1
·
1 Parent(s): d58060f

Update LLM model configuration

Browse files
Files changed (2) hide show
  1. __pycache__/app.cpython-310.pyc +0 -0
  2. app.py +8 -8
__pycache__/app.cpython-310.pyc ADDED
Binary file (7.95 kB). View file
 
app.py CHANGED
@@ -13,8 +13,8 @@ from llama_cpp import Llama
13
 
14
  TTS_MODEL_ID = os.getenv("TTS_MODEL_ID", "Qwen/Qwen3-TTS-12Hz-0.6B-CustomVoice")
15
 
16
- LLM_REPO_ID = os.getenv("LLM_REPO_ID", "Qwen/Qwen2.5-1.5B-Instruct-GGUF")
17
- LLM_FILENAME = os.getenv("LLM_FILENAME", "qwen2.5-1.5b-instruct-q4_k_m.gguf")
18
 
19
  tts_model = None
20
  llm_model = None
@@ -146,7 +146,7 @@ def get_llm_model():
146
  llm_model = Llama.from_pretrained(
147
  repo_id=LLM_REPO_ID,
148
  filename=LLM_FILENAME,
149
- n_ctx=2048,
150
  n_threads=max(2, os.cpu_count() or 2),
151
  n_gpu_layers=-1 if torch.cuda.is_available() else 0,
152
  verbose=True,
@@ -195,7 +195,7 @@ Rules:
195
  {"role": "user", "content": user_prompt},
196
  ],
197
  temperature=0.2,
198
- max_tokens=256,
199
  )
200
 
201
  content = result["choices"][0]["message"]["content"].strip()
@@ -271,10 +271,10 @@ with gr.Blocks(
271
  Current milestone: llama.cpp intent generation + Qwen3-TTS voice synthesis.
272
  </p>
273
  <div class="badge-row">
274
- <span class="badge">🦙 llama.cpp</span>
275
- <span class="badge">🔌 local-first</span>
276
- <span class="badge">🎨 custom Gradio UI</span>
277
- <span class="badge">🗣️ expressive TTS</span>
278
  </div>
279
  </section>
280
  """
 
13
 
14
  TTS_MODEL_ID = os.getenv("TTS_MODEL_ID", "Qwen/Qwen3-TTS-12Hz-0.6B-CustomVoice")
15
 
16
+ LLM_REPO_ID = os.getenv("LLM_REPO_ID", "Qwen/Qwen2.5-0.5B-Instruct-GGUF")
17
+ LLM_FILENAME = os.getenv("LLM_FILENAME", "qwen2.5-0.5b-instruct-q4_k_m.gguf")
18
 
19
  tts_model = None
20
  llm_model = None
 
146
  llm_model = Llama.from_pretrained(
147
  repo_id=LLM_REPO_ID,
148
  filename=LLM_FILENAME,
149
+ n_ctx=1024,
150
  n_threads=max(2, os.cpu_count() or 2),
151
  n_gpu_layers=-1 if torch.cuda.is_available() else 0,
152
  verbose=True,
 
195
  {"role": "user", "content": user_prompt},
196
  ],
197
  temperature=0.2,
198
+ max_tokens=96,
199
  )
200
 
201
  content = result["choices"][0]["message"]["content"].strip()
 
271
  Current milestone: llama.cpp intent generation + Qwen3-TTS voice synthesis.
272
  </p>
273
  <div class="badge-row">
274
+ <span class="badge">llama.cpp</span>
275
+ <span class="badge">local-first</span>
276
+ <span class="badge">custom Gradio UI</span>
277
+ <span class="badge">expressive TTS</span>
278
  </div>
279
  </section>
280
  """