ayush-vatsal/description_to_caption
Viewer • Updated • 121 • 9
How to use ayush-vatsal/caption_qlora_finetune with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("vilsonrodrigues/falcon-7b-instruct-sharded")
model = PeftModel.from_pretrained(base_model, "ayush-vatsal/caption_qlora_finetune")The following bitsandbytes quantization config was used during training: