OEvortex
/

BabyMistral

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Aarifkhan commited on Aug 20

Commit

b34fe72

•

1 Parent(s): b9459fe

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -52,8 +52,8 @@ To use BabyMistral with the Hugging Face Transformers library:
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model = AutoModelForCausalLM.from_pretrained("Aarifkhan/BabyMistral")
-tokenizer = AutoTokenizer.from_pretrained("Aarifkhan/BabyMistral")
 # Define the chat input
 chat = [

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model = AutoModelForCausalLM.from_pretrained("OEvortex/BabyMistral")
+tokenizer = AutoTokenizer.from_pretrained("OEvortex/BabyMistral")
 # Define the chat input
 chat = [