how do you use this model
#2
by erde9 - opened
How do you use this model? Do you have any example?
or you can load the model directly:
Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-generation", model="Rahulholla/mistral-stock-model")
Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("Rahulholla/mistral-stock-model")
model = AutoModelForCausalLM.from_pretrained("Rahulholla/mistral-stock-model")
Rahulholla changed discussion status to closed