Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4base_model = AutoModelForCausalLM.from_pretrained("BASE_MODEL_NAME")
5tokenizer = AutoTokenizer.from_pretrained("BASE_MODEL_NAME")
6
7model = PeftModel.from_pretrained(base_model, "pnkjkhani/tiny-no-robot")