guardrail commited on
Commit
413472c
1 Parent(s): 0b9e1a4

Create README.md

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