Transformers
GGUF
English
mav23 commited on
Commit
4a5dd31
·
verified ·
1 Parent(s): bba1228

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. .gitattributes +1 -0
  2. README.md +188 -0
  3. dolly-v2-7b.Q4_0.gguf +3 -0
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ dolly-v2-7b.Q4_0.gguf filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,188 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ language:
4
+ - en
5
+ library_name: transformers
6
+ inference: false
7
+ datasets:
8
+ - databricks/databricks-dolly-15k
9
+ ---
10
+ # dolly-v2-7b Model Card
11
+ ## Summary
12
+
13
+ Databricks' `dolly-v2-7b`, an instruction-following large language model trained on the Databricks machine learning platform
14
+ that is licensed for commercial use. Based on `pythia-6.9b`, Dolly is trained on ~15k instruction/response fine tuning records
15
+ [`databricks-dolly-15k`](https://github.com/databrickslabs/dolly/tree/master/data) generated
16
+ by Databricks employees in capability domains from the InstructGPT paper, including brainstorming, classification, closed QA, generation,
17
+ information extraction, open QA and summarization. `dolly-v2-7b` is not a state-of-the-art model, but does exhibit surprisingly
18
+ high quality instruction following behavior not characteristic of the foundation model on which it is based.
19
+
20
+ Dolly v2 is also available in these other models sizes:
21
+
22
+ * [dolly-v2-12b](https://huggingface.co/databricks/dolly-v2-12b), a 12 billion parameter based on `pythia-12b`
23
+ * [dolly-v2-3b](https://huggingface.co/databricks/dolly-v2-3b), a 2.8 billion parameter based on `pythia-2.8b`
24
+
25
+ Please refer to the [dolly GitHub repo](https://github.com/databrickslabs/dolly#getting-started-with-response-generation) for tips on
26
+ running inference for various GPU configurations.
27
+
28
+ **Owner**: Databricks, Inc.
29
+
30
+ ## Model Overview
31
+ `dolly-v2-7b` is a 6.9 billion parameter causal language model created by [Databricks](https://databricks.com/) that is derived from
32
+ [EleutherAI's](https://www.eleuther.ai/) [Pythia-6.9b](https://huggingface.co/EleutherAI/pythia-6.9b) and fine-tuned
33
+ on a [~15K record instruction corpus](https://github.com/databrickslabs/dolly/tree/master/data) generated by Databricks employees and released under a permissive license (CC-BY-SA)
34
+
35
+ ## Usage
36
+
37
+ To use the model with the `transformers` library on a machine with GPUs, first make sure you have the `transformers` and `accelerate` libraries installed.
38
+ In a Databricks notebook you could run:
39
+
40
+ ```python
41
+ %pip install "accelerate>=0.16.0,<1" "transformers[torch]>=4.28.1,<5" "torch>=1.13.1,<2"
42
+ ```
43
+
44
+ The instruction following pipeline can be loaded using the `pipeline` function as shown below. This loads a custom `InstructionTextGenerationPipeline`
45
+ found in the model repo [here](https://huggingface.co/databricks/dolly-v2-3b/blob/main/instruct_pipeline.py), which is why `trust_remote_code=True` is required.
46
+ Including `torch_dtype=torch.bfloat16` is generally recommended if this type is supported in order to reduce memory usage. It does not appear to impact output quality.
47
+ It is also fine to remove it if there is sufficient memory.
48
+
49
+ ```python
50
+ import torch
51
+ from transformers import pipeline
52
+
53
+ generate_text = pipeline(model="databricks/dolly-v2-7b", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
54
+ ```
55
+
56
+ You can then use the pipeline to answer instructions:
57
+
58
+ ```python
59
+ res = generate_text("Explain to me the difference between nuclear fission and fusion.")
60
+ print(res[0]["generated_text"])
61
+ ```
62
+
63
+ Alternatively, if you prefer to not use `trust_remote_code=True` you can download [instruct_pipeline.py](https://huggingface.co/databricks/dolly-v2-3b/blob/main/instruct_pipeline.py),
64
+ store it alongside your notebook, and construct the pipeline yourself from the loaded model and tokenizer:
65
+
66
+ ```python
67
+ import torch
68
+ from instruct_pipeline import InstructionTextGenerationPipeline
69
+ from transformers import AutoModelForCausalLM, AutoTokenizer
70
+
71
+ tokenizer = AutoTokenizer.from_pretrained("databricks/dolly-v2-7b", padding_side="left")
72
+ model = AutoModelForCausalLM.from_pretrained("databricks/dolly-v2-7b", device_map="auto", torch_dtype=torch.bfloat16)
73
+
74
+ generate_text = InstructionTextGenerationPipeline(model=model, tokenizer=tokenizer)
75
+ ```
76
+
77
+ ### LangChain Usage
78
+
79
+ To use the pipeline with LangChain, you must set `return_full_text=True`, as LangChain expects the full text to be returned
80
+ and the default for the pipeline is to only return the new text.
81
+
82
+ ```python
83
+ import torch
84
+ from transformers import pipeline
85
+
86
+ generate_text = pipeline(model="databricks/dolly-v2-7b", torch_dtype=torch.bfloat16,
87
+ trust_remote_code=True, device_map="auto", return_full_text=True)
88
+ ```
89
+
90
+ You can create a prompt that either has only an instruction or has an instruction with context:
91
+
92
+ ```python
93
+ from langchain import PromptTemplate, LLMChain
94
+ from langchain.llms import HuggingFacePipeline
95
+
96
+ # template for an instrution with no input
97
+ prompt = PromptTemplate(
98
+ input_variables=["instruction"],
99
+ template="{instruction}")
100
+
101
+ # template for an instruction with input
102
+ prompt_with_context = PromptTemplate(
103
+ input_variables=["instruction", "context"],
104
+ template="{instruction}\n\nInput:\n{context}")
105
+
106
+ hf_pipeline = HuggingFacePipeline(pipeline=generate_text)
107
+
108
+ llm_chain = LLMChain(llm=hf_pipeline, prompt=prompt)
109
+ llm_context_chain = LLMChain(llm=hf_pipeline, prompt=prompt_with_context)
110
+ ```
111
+
112
+ Example predicting using a simple instruction:
113
+
114
+ ```python
115
+ print(llm_chain.predict(instruction="Explain to me the difference between nuclear fission and fusion.").lstrip())
116
+ ```
117
+
118
+ Example predicting using an instruction with context:
119
+
120
+ ```python
121
+ context = """George Washington (February 22, 1732[b] - December 14, 1799) was an American military officer, statesman,
122
+ and Founding Father who served as the first president of the United States from 1789 to 1797."""
123
+
124
+ print(llm_context_chain.predict(instruction="When was George Washington president?", context=context).lstrip())
125
+ ```
126
+
127
+
128
+ ## Known Limitations
129
+
130
+ ### Performance Limitations
131
+ **`dolly-v2-7b` is not a state-of-the-art generative language model** and, though quantitative benchmarking is ongoing, is not designed to perform
132
+ competitively with more modern model architectures or models subject to larger pretraining corpuses.
133
+
134
+ The Dolly model family is under active development, and so any list of shortcomings is unlikely to be exhaustive, but we include known limitations and misfires here as a means to document and share our preliminary findings with the community.
135
+ In particular, `dolly-v2-7b` struggles with: syntactically complex prompts, programming problems, mathematical operations, factual errors,
136
+ dates and times, open-ended question answering, hallucination, enumerating lists of specific length, stylistic mimicry, having a sense of humor, etc.
137
+ Moreover, we find that `dolly-v2-7b` does not have some capabilities, such as well-formatted letter writing, present in the original model.
138
+
139
+ ### Dataset Limitations
140
+ Like all language models, `dolly-v2-7b` reflects the content and limitations of its training corpuses.
141
+
142
+ - **The Pile**: GPT-J's pre-training corpus contains content mostly collected from the public internet, and like most web-scale datasets,
143
+ it contains content many users would find objectionable. As such, the model is likely to reflect these shortcomings, potentially overtly
144
+ in the case it is explicitly asked to produce objectionable content, and sometimes subtly, as in the case of biased or harmful implicit
145
+ associations.
146
+
147
+ - **`databricks-dolly-15k`**: The training data on which `dolly-v2-7b` is instruction tuned represents natural language instructions generated
148
+ by Databricks employees during a period spanning March and April 2023 and includes passages from Wikipedia as references passages
149
+ for instruction categories like closed QA and summarization. To our knowledge it does not contain obscenity, intellectual property or
150
+ personally identifying information about non-public figures, but it may contain typos and factual errors.
151
+ The dataset may also reflect biases found in Wikipedia. Finally, the dataset likely reflects
152
+ the interests and semantic choices of Databricks employees, a demographic which is not representative of the global population at large.
153
+
154
+ Databricks is committed to ongoing research and development efforts to develop helpful, honest and harmless AI technologies that
155
+ maximize the potential of all individuals and organizations.
156
+
157
+ ### Benchmark Metrics
158
+
159
+ Below you'll find various models benchmark performance on the [EleutherAI LLM Evaluation Harness](https://github.com/EleutherAI/lm-evaluation-harness);
160
+ model results are sorted by geometric mean to produce an intelligible ordering. As outlined above, these results demonstrate that `dolly-v2-7b` is not state of the art,
161
+ and in fact underperforms `dolly-v1-6b` in some evaluation benchmarks. We believe this owes to the composition and size of the underlying fine tuning datasets,
162
+ but a robust statement as to the sources of these variations requires further study.
163
+
164
+ | model | openbookqa | arc_easy | winogrande | hellaswag | arc_challenge | piqa | boolq | gmean |
165
+ | --------------------------------- | ------------ | ---------- | ------------ | ----------- | --------------- | -------- | -------- | ---------|
166
+ | EleutherAI/pythia-2.8b | 0.348 | 0.585859 | 0.589582 | 0.591217 | 0.323379 | 0.73395 | 0.638226 | 0.523431 |
167
+ | EleutherAI/pythia-6.9b | 0.368 | 0.604798 | 0.608524 | 0.631548 | 0.343857 | 0.761153 | 0.6263 | 0.543567 |
168
+ | databricks/dolly-v2-3b | 0.384 | 0.611532 | 0.589582 | 0.650767 | 0.370307 | 0.742655 | 0.575535 | 0.544886 |
169
+ | EleutherAI/pythia-12b | 0.364 | 0.627104 | 0.636148 | 0.668094 | 0.346416 | 0.760065 | 0.673394 | 0.559676 |
170
+ | EleutherAI/gpt-j-6B | 0.382 | 0.621633 | 0.651144 | 0.662617 | 0.363481 | 0.761153 | 0.655963 | 0.565936 |
171
+ | databricks/dolly-v2-12b | 0.408 | 0.63931 | 0.616417 | 0.707927 | 0.388225 | 0.757889 | 0.568196 | 0.56781 |
172
+ | databricks/dolly-v2-7b | 0.392 | 0.633838 | 0.607735 | 0.686517 | 0.406997 | 0.750816 | 0.644037 | 0.573487 |
173
+ | databricks/dolly-v1-6b | 0.41 | 0.62963 | 0.643252 | 0.676758 | 0.384812 | 0.773667 | 0.687768 | 0.583431 |
174
+ | EleutherAI/gpt-neox-20b | 0.402 | 0.683923 | 0.656669 | 0.7142 | 0.408703 | 0.784004 | 0.695413 | 0.602236 |
175
+
176
+ # Citation
177
+
178
+ ```
179
+ @online{DatabricksBlog2023DollyV2,
180
+ author = {Mike Conover and Matt Hayes and Ankit Mathur and Jianwei Xie and Jun Wan and Sam Shah and Ali Ghodsi and Patrick Wendell and Matei Zaharia and Reynold Xin},
181
+ title = {Free Dolly: Introducing the World's First Truly Open Instruction-Tuned LLM},
182
+ year = {2023},
183
+ url = {https://www.databricks.com/blog/2023/04/12/dolly-first-open-commercially-viable-instruction-tuned-llm},
184
+ urldate = {2023-06-30}
185
+ }
186
+ ```
187
+
188
+ # Happy Hacking!
dolly-v2-7b.Q4_0.gguf ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf7765c0549cd903d5ee9f467878391cbc64837a425baf63259629af9e73d650
3
+ size 3917303232