ecyht2's picture
Create app.py
bebd6a0 verified
raw
history blame
256 Bytes
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained(
"cognitivecomputations/dolphin-2.6-mistral-7b"
)
model = AutoModelForCausalLM.from_pretrained(
"cognitivecomputations/dolphin-2.6-mistral-7b"
)