vigostral-chat / setup.py
ilhooq's picture
Try to download the model at build time
715c978
raw
history blame
220 Bytes
from ctransformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("TheBloke/Vigostral-7B-Chat-GGUF",
model_file="vigostral-7b-chat.Q4_K_M.gguf",
model_type="mistral",
threads = 3
)