AdamDE's picture
Upload folder using huggingface_hub
4fb90d5 verified
|
raw
history blame
667 Bytes
metadata
library_name: peft
base_model: TinyLlama/TinyLlama-1.1B-Chat-v1.0
pipeline_tag: text-generation
tags:
  - lora
  - adapters
  - tinyllama

TinyLlama YouTube Replies (LoRA)

LoRA adapters to fine-tune TinyLlama-1.1B-Chat-v1.0 for generating concise, friendly replies to AI/ML YouTube comments.

Usage

from transformers import AutoTokenizer, AutoModelForCausalLM
from peft import PeftModel
base = AutoModelForCausalLM.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
tok = AutoTokenizer.from_pretrained("your-username/tinyllama-youtube-replies-lora")
model = PeftModel.from_pretrained(base, "your-username/tinyllama-youtube-replies-lora")