EryriLabs commited on
Commit
4f61acb
1 Parent(s): 7e2124c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +18 -0
README.md CHANGED
@@ -26,6 +26,24 @@ Draig-Fach-v0.1 is an instruction fine-tuned small language model based on the M
26
  - **License:** cc-by-nc-4.0
27
  - **Finetuned from model:** mistralai/Mistral-7B-Instruct-v0.2
28
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
29
  ## Uses
30
 
31
  Draig-Fach-v0.1 is intended for:
 
26
  - **License:** cc-by-nc-4.0
27
  - **Finetuned from model:** mistralai/Mistral-7B-Instruct-v0.2
28
 
29
+
30
+
31
+ ## How to use
32
+
33
+ You can use this model directly with a Hugging Face pipeline:
34
+ ```python
35
+
36
+ from transformers import pipeline, Conversation
37
+ import torch
38
+
39
+ base_model_name = "EryriLabs/Draig-Fach-v0.1"
40
+ chatbot = pipeline("conversational", model=base_model_name, torch_dtype=torch.float16, device_map="auto")
41
+ conversation = Conversation("Sut wyt ti?")
42
+ conversation = chatbot(conversation)
43
+ print(conversation.messages[-1]["content"])
44
+
45
+ ```
46
+
47
  ## Uses
48
 
49
  Draig-Fach-v0.1 is intended for: