Commit ·
bb615f1
1
Parent(s): d58060f
Update LLM model configuration
Browse files- __pycache__/app.cpython-310.pyc +0 -0
- 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-
|
| 17 |
-
LLM_FILENAME = os.getenv("LLM_FILENAME", "qwen2.5-
|
| 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=
|
| 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=
|
| 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">
|
| 275 |
-
<span class="badge">
|
| 276 |
-
<span class="badge">
|
| 277 |
-
<span class="badge">
|
| 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 |
"""
|