How to use bilkultheek/ColdLLamaLite with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("ahxt/LiteLlama-460M-1T") model = PeftModel.from_pretrained(base_model, "bilkultheek/ColdLLamaLite")
