pankajmathur commited on
Commit
9b22310
1 Parent(s): 0ea5bec

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -31,8 +31,13 @@ You are Dolphin, a helpful AI assistant.<|im_end|>
31
 
32
  ChatML prompt template is available as a chat template, which means you can format messages using the tokenizer.apply_chat_template() method:
33
  ```
 
 
 
 
 
34
  messages = [
35
- {"role": "system", "content": "You are BrokeLlama, a helpful AI assistant."},
36
  {"role": "user", "content": "Hello BrokenLlama, what can you do for me?"}
37
  ]
38
 
 
31
 
32
  ChatML prompt template is available as a chat template, which means you can format messages using the tokenizer.apply_chat_template() method:
33
  ```
34
+ from transformers import AutoConfig, AutoModel, AutoTokenizer
35
+ config = AutoConfig.from_pretrained("investbrainsorg/BrokenLlama-3-8b")
36
+ model = AutoModel.from_pretrained("investbrainsorg/BrokenLlama-3-8b")
37
+ tokenizer = AutoTokenizer.from_pretrained("investbrainsorg/BrokenLlama-3-8b")
38
+
39
  messages = [
40
+ {"role": "system", "content": "You are BrokenLlama, a helpful AI assistant."},
41
  {"role": "user", "content": "Hello BrokenLlama, what can you do for me?"}
42
  ]
43