Skip to content

Commit

Permalink
Update default model to `bartowski/Qwen2.5-7B-Instruct-GGUF/Qwen2.5-7…
Browse files Browse the repository at this point in the history
…B-Instruct-Q8_0.gguf`
  • Loading branch information
daavoo committed Feb 6, 2025
1 parent d8293c2 commit a8cfab4
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 3 deletions.
2 changes: 1 addition & 1 deletion demo/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
@st.cache_resource
def load_model():
return load_llama_cpp_model(
"bartowski/Qwen2.5-3B-Instruct-GGUF/Qwen2.5-3B-Instruct-f16.gguf"
"bartowski/Qwen2.5-7B-Instruct-GGUF/Qwen2.5-7B-Instruct-Q8_0.gguf"
)


Expand Down
2 changes: 1 addition & 1 deletion example_data/config.yaml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
question: "What optimizer was used to train the model?"
input_file: example_data/1706.03762v7.pdf
output_dir: example_outputs/1706.03762v7.pdf
model: "bartowski/Qwen2.5-3B-Instruct-GGUF/Qwen2.5-3B-Instruct-f16.gguf"
model: "bartowski/Qwen2.5-7B-Instruct-GGUF/Qwen2.5-7B-Instruct-Q8_0.gguf"
find_prompt : |
You are given two pieces of information:
1. A list of valid section names.
Expand Down
3 changes: 2 additions & 1 deletion src/structured_qa/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ def structured_qa(
input_file: str | None = None,
output_dir: str | None = None,
model: str
| None = "bartowski/Qwen2.5-3B-Instruct-GGUF/Qwen2.5-3B-Instruct-f16.gguf",
| None = "bartowski/Qwen2.5-7B-Instruct-GGUF/Qwen2.5-7B-Instruct-Q8_0.gguf",
find_prompt: str = FIND_PROMPT,
answer_prompt: str = ANSWER_PROMPT,
from_config: str | None = None,
Expand All @@ -39,6 +39,7 @@ def structured_qa(
```
model: Model identifier formatted as `owner/repo/file`.
Must be hosted at the HuggingFace Hub in GGUF format.
Defaults to bartowski/Qwen2.5-7B-Instruct-GGUF/Qwen2.5-7B-Instruct-Q8_0.gguf.
question: The question to answer.
find_prompt: The prompt for finding the section.
Expand Down

0 comments on commit a8cfab4

Please sign in to comment.