File size: 292 Bytes
77e0b1e
 
 
 
 
1
2
3
4
5
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("loubnabnl/santacoder-code-to-text", trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained("loubnabnl/santacoder-code-to-text", trust_remote_code=True)