granite / app.py
JohnBoot's picture
Update app.py
2a55f6c verified
raw
history blame contribute delete
250 Bytes
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("ibm-granite/granite-3b-code-instruct")
model = AutoModelForCausalLM.from_pretrained("ibm-granite/granite-3b-code-instruct")