How to use michaelc0des/gemma-direct-commentary with PEFT:
from peft import PeftModel from transformers import AutoModelForSeq2SeqLM base_model = AutoModelForSeq2SeqLM.from_pretrained("google/t5gemma-b-b-prefixlm-it") model = PeftModel.from_pretrained(base_model, "michaelc0des/gemma-direct-commentary")
How to use michaelc0des/gemma-direct-commentary with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("michaelc0des/gemma-direct-commentary", device_map="auto")
