# 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") |