doberst commited on
Commit
8cf9eb1
·
verified ·
1 Parent(s): da107e5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -17
README.md CHANGED
@@ -1,39 +1,46 @@
1
  ---
2
- license: llama2
 
3
  ---
4
 
5
- # Model Card for Model ID
6
 
7
  <!-- Provide a quick summary of what the model is/does. -->
8
 
9
- dragon-llama-7b-v0 part of the dRAGon ("Delivering RAG On ...") model series, RAG-instruct trained on top of a LLama-2 base model.
10
 
11
  DRAGON models have been fine-tuned with the specific objective of fact-based question-answering over complex business and legal documents with an emphasis on reducing hallucinations and providing short, clear answers for workflow automation.
12
 
 
13
  ### Benchmark Tests
14
 
15
  Evaluated against the benchmark test: [RAG-Instruct-Benchmark-Tester](https://www.huggingface.co/datasets/llmware/rag_instruct_benchmark_tester)
16
- Average of 2 Test Runs with 1 point for correct answer, 0.5 point for partial correct or blank / NF, 0.0 points for incorrect, and -1 points for hallucinations.
17
-
18
- --**Accuracy Score**: **97.25** correct out of 100
19
- --Not Found Classification: 92.50%
20
- --Boolean: 95.00%
21
- --Math/Logic: 63.75%
22
- --Complex Questions (1-5): 3 (Medium)
23
- --Summarization Quality (1-5): 3 (Coherent, extractive)
24
  --Hallucinations: No hallucinations observed in test runs.
25
 
26
- For test run results (and good indicator of target use cases), please see the files ("core_rag_test" and "answer_sheet" in this repo).
 
 
 
 
 
27
 
28
  ### Model Description
29
 
30
  <!-- Provide a longer summary of what this model is. -->
31
 
32
  - **Developed by:** llmware
33
- - **Model type:** LLama-2
34
  - **Language(s) (NLP):** English
35
- - **License:** LLama 2 Community License Agreement
36
- - **Finetuned from model:** Llama-2-7B-Base
37
 
38
 
39
  ### Direct Use
@@ -59,8 +66,8 @@ Any model can provide inaccurate or incomplete information, and should be used i
59
  The fastest way to get started with dRAGon is through direct import in transformers:
60
 
61
  from transformers import AutoTokenizer, AutoModelForCausalLM
62
- tokenizer = AutoTokenizer.from_pretrained("dragon-llama-7b-v0")
63
- model = AutoModelForCausalLM.from_pretrained("dragon-llama-7b-v0")
64
 
65
  Please refer to the generation_test .py files in the Files repository, which includes 200 samples and script to test the model. The **generation_test_llmware_script.py** includes built-in llmware capabilities for fact-checking, as well as easy integration with document parsing and actual retrieval to swap out the test set for RAG workflow consisting of business documents.
66
 
 
1
  ---
2
+ license: apache-2.0
3
+ inference: false
4
  ---
5
 
6
+ # DRAGON-MISTRAL-0.3
7
 
8
  <!-- Provide a quick summary of what the model is/does. -->
9
 
10
+ dragon-mistral-0.3 is part of the dRAGon ("Delivering RAG On ...") model series, RAG-instruct trained on top of a Mistral 7b (0.3) base model.
11
 
12
  DRAGON models have been fine-tuned with the specific objective of fact-based question-answering over complex business and legal documents with an emphasis on reducing hallucinations and providing short, clear answers for workflow automation.
13
 
14
+
15
  ### Benchmark Tests
16
 
17
  Evaluated against the benchmark test: [RAG-Instruct-Benchmark-Tester](https://www.huggingface.co/datasets/llmware/rag_instruct_benchmark_tester)
18
+ 1 Test Run (with temperature = 0.0 and sample = False) with 1 point for correct answer, 0.5 point for partial correct or blank / NF, 0.0 points for incorrect, and -1 points for hallucinations.
19
+
20
+ --**Accuracy Score**: **99.5** correct out of 100
21
+ --Not Found Classification: 95.0%
22
+ --Boolean: 82.5%
23
+ --Math/Logic: 67.5%
24
+ --Complex Questions (1-5): 4 (Above Average - multiple-choice, causal)
25
+ --Summarization Quality (1-5): 4 (Above Average)
26
  --Hallucinations: No hallucinations observed in test runs.
27
 
28
+ For test run results (and good indicator of target use cases), please see the files ("core_rag_test" and "answer_sheet" in this repo).
29
+
30
+ Please note that these test results were achieved using the 4_K_M quantized version of this model - [dragon-mistral-0.3-gguf](https://wwww.huggingface.co/llmware/dragon-mistral-0.3-gguf).
31
+
32
+ Note: compare results with [dragon-mistral-7b](https://www.huggingface.co/llmware/dragon-mistral-7b-v0).
33
+
34
 
35
  ### Model Description
36
 
37
  <!-- Provide a longer summary of what this model is. -->
38
 
39
  - **Developed by:** llmware
40
+ - **Model type:** mistral
41
  - **Language(s) (NLP):** English
42
+ - **License:** Apache 2.0
43
+ - **Finetuned from model:** Mistral-7b-base-0.3
44
 
45
 
46
  ### Direct Use
 
66
  The fastest way to get started with dRAGon is through direct import in transformers:
67
 
68
  from transformers import AutoTokenizer, AutoModelForCausalLM
69
+ tokenizer = AutoTokenizer.from_pretrained("dragon-mistral-0.3")
70
+ model = AutoModelForCausalLM.from_pretrained("dragon-mistral-0.3")
71
 
72
  Please refer to the generation_test .py files in the Files repository, which includes 200 samples and script to test the model. The **generation_test_llmware_script.py** includes built-in llmware capabilities for fact-checking, as well as easy integration with document parsing and actual retrieval to swap out the test set for RAG workflow consisting of business documents.
73