File tree 3 files changed +4
-3
lines changed
3 files changed +4
-3
lines changed Original file line number Diff line number Diff line change 13
13
@st .cache_resource
14
14
def load_model ():
15
15
return load_llama_cpp_model (
16
- "bartowski/Qwen2.5-3B -Instruct-GGUF/Qwen2.5-3B -Instruct-f16 .gguf"
16
+ "bartowski/Qwen2.5-7B -Instruct-GGUF/Qwen2.5-7B -Instruct-Q8_0 .gguf"
17
17
)
18
18
19
19
Original file line number Diff line number Diff line change 1
1
question : " What optimizer was used to train the model?"
2
2
input_file : example_data/1706.03762v7.pdf
3
3
output_dir : example_outputs/1706.03762v7.pdf
4
- model : " bartowski/Qwen2.5-3B -Instruct-GGUF/Qwen2.5-3B -Instruct-f16 .gguf"
4
+ model : " bartowski/Qwen2.5-7B -Instruct-GGUF/Qwen2.5-7B -Instruct-Q8_0 .gguf"
5
5
find_prompt : |
6
6
You are given two pieces of information:
7
7
1. A list of valid section names.
Original file line number Diff line number Diff line change @@ -16,7 +16,7 @@ def structured_qa(
16
16
input_file : str | None = None ,
17
17
output_dir : str | None = None ,
18
18
model : str
19
- | None = "bartowski/Qwen2.5-3B -Instruct-GGUF/Qwen2.5-3B -Instruct-f16 .gguf" ,
19
+ | None = "bartowski/Qwen2.5-7B -Instruct-GGUF/Qwen2.5-7B -Instruct-Q8_0 .gguf" ,
20
20
find_prompt : str = FIND_PROMPT ,
21
21
answer_prompt : str = ANSWER_PROMPT ,
22
22
from_config : str | None = None ,
@@ -39,6 +39,7 @@ def structured_qa(
39
39
```
40
40
model: Model identifier formatted as `owner/repo/file`.
41
41
Must be hosted at the HuggingFace Hub in GGUF format.
42
+ Defaults to bartowski/Qwen2.5-7B-Instruct-GGUF/Qwen2.5-7B-Instruct-Q8_0.gguf.
42
43
question: The question to answer.
43
44
find_prompt: The prompt for finding the section.
44
45
You can’t perform that action at this time.
0 commit comments