Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3# Load model and tokenizer
4model_name = "U4RASD/dalla-gemma-it" # or "U4RASD/dalla-llama-it"
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForCausalLM.from_pretrained(model_name)