Commit
•
9e2673b
1
Parent(s):
ee79699
Update README.md (#3)
Browse files- Update README.md (66b72c5f5592e3d6302da1f2cffa25862496fedb)
Co-authored-by: dumball <[email protected]>
README.md
CHANGED
|
@@ -281,7 +281,7 @@ Note: To use function calling, you should see the github repo above.
|
|
| 281 |
|
| 282 |
import torch
|
| 283 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 284 |
-
from transformers import LlamaTokenizer,
|
| 285 |
import bitsandbytes, flash_attn
|
| 286 |
|
| 287 |
tokenizer = LlamaTokenizer.from_pretrained('NousResearch/Hermes-2-Pro-Mistral-7B', trust_remote_code=True)
|
|
|
|
| 281 |
|
| 282 |
import torch
|
| 283 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 284 |
+
from transformers import LlamaTokenizer, MistralForCausalLM
|
| 285 |
import bitsandbytes, flash_attn
|
| 286 |
|
| 287 |
tokenizer = LlamaTokenizer.from_pretrained('NousResearch/Hermes-2-Pro-Mistral-7B', trust_remote_code=True)
|