Views
No views yet
mistral_inference, a mistral-demo CLI command should be available in your environment.[!IMPORTANT] NOTE: Until a new release has been made, you need to install transformers from source:pip install git+https://github.com/huggingface/transformers.git
transformers to generate text, you can do something like this.1from transformers import AutoModelForCausalLM, AutoTokenizer
2model_id = "EpistemeAI/Fireball-MathMistral-Nemo-Base-2407"
3tokenizer = AutoTokenizer.from_pretrained(model_id)
4model = AutoModelForCausalLM.from_pretrained(model_id)
5inputs = tokenizer("Hello my name is", return_tensors="pt")
6outputs = model.generate(**inputs, max_new_tokens=20)
7print(tokenizer.decode(outputs[0], skip_special_tokens=True))[!TIP] Unlike previous Mistral models, Mistral Nemo requires smaller temperatures. We recommend to use a temperature of 0.3.
Mistral-Nemo-Base-2407 is a pretrained base model and therefore does not have any moderation mechanisms.