guardrail commited on
Commit
86e6783
·
1 Parent(s): d708b67

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +21 -0
README.md ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ datasets:
4
+ - databricks/databricks-dolly-15k
5
+ - timdettmers/openassistant-guanaco
6
+ ---
7
+ ---
8
+
9
+ Model that is fine-tuned in 4-bit precision using QLoRA on [timdettmers/openassistant-guanaco](https://huggingface.co/datasets/timdettmers/openassistant-guanaco) and [databricks/databricks-dolly-15k](https://huggingface.co/datasets/databricks/databricks-dolly-15k). Sharded as well to be used on a free Google Colab instance.
10
+
11
+ It can be easily imported using the `AutoModelForCausalLM` class from `transformers`:
12
+
13
+ ```
14
+ from transformers import AutoModelForCausalLM, AutoTokenizer
15
+
16
+ model = AutoModelForCausalLM.from_pretrained(
17
+ "guardrail/llama-2-7b-guanaco-dolly-8bit-sharded",
18
+ load_in_8bit=True)
19
+ tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
20
+
21
+ ```