cminja commited on
Commit
509c667
1 Parent(s): 4505ea1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +27 -3
README.md CHANGED
@@ -1,3 +1,27 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ ---
4
+
5
+
6
+ ```python
7
+ !pip install -qU transformers accelerate
8
+
9
+ from transformers import AutoTokenizer
10
+ import transformers
11
+ import torch
12
+
13
+ model = "cminja/dolphin-2.8-mistral-11b-v02-sft-code-fn-calling"
14
+ messages = [{"role": "user", "content": "What is a large language model?"}]
15
+
16
+ tokenizer = AutoTokenizer.from_pretrained(model)
17
+ prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
18
+ pipeline = transformers.pipeline(
19
+ "text-generation",
20
+ model=model,
21
+ torch_dtype=torch.float16,
22
+ device_map="auto",
23
+ )
24
+
25
+ outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
26
+ print(outputs[0]["generated_text"])
27
+ ```