birgermoell commited on
Commit
d3e86d8
·
verified ·
1 Parent(s): 19b0aab

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -3
README.md CHANGED
@@ -13,15 +13,17 @@ tags:
13
 
14
  ## How to use
15
 
16
-
17
  !pip install -qU transformers accelerate
 
18
 
 
19
  from transformers import AutoTokenizer
20
  import transformers
21
  import torch
22
 
23
- model = "timpal0l/BeagleCatMunin2"
24
- messages = [{"role": "user", "content": "What is a large language model?"}]
25
 
26
  tokenizer = AutoTokenizer.from_pretrained(model)
27
  prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
@@ -34,6 +36,7 @@ pipeline = transformers.pipeline(
34
 
35
  outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
36
  print(outputs[0]["generated_text"])
 
37
 
38
  This is a merge of pre-trained language models created using [mergekit](https://github.com/cg123/mergekit).
39
 
 
13
 
14
  ## How to use
15
 
16
+ ```bash
17
  !pip install -qU transformers accelerate
18
+ ```
19
 
20
+ ```python
21
  from transformers import AutoTokenizer
22
  import transformers
23
  import torch
24
 
25
+ model = "birgermoell/eir"
26
+ messages = [{"role": "user", "content": "Vad är symptom lunginflammation?"}]
27
 
28
  tokenizer = AutoTokenizer.from_pretrained(model)
29
  prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
 
36
 
37
  outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
38
  print(outputs[0]["generated_text"])
39
+ ```
40
 
41
  This is a merge of pre-trained language models created using [mergekit](https://github.com/cg123/mergekit).
42