How to use blankenshipdanielj/gitgraph-model with PEFT:
from peft import PeftModel from transformers import AutoModelForSeq2SeqLM base_model = AutoModelForSeq2SeqLM.from_pretrained("Salesforce/codet5p-220m") model = PeftModel.from_pretrained(base_model, "blankenshipdanielj/gitgraph-model")
How to use blankenshipdanielj/gitgraph-model with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("blankenshipdanielj/gitgraph-model", device_map="auto")
