vicgalle/alpaca-gpt4
Viewer • Updated • 52k • 4.49k • 324
How to use zjudai/flowertune-general-nlp-lora-tinyllama-1.1b-chat-v1.0 with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
model = PeftModel.from_pretrained(base_model, "zjudai/flowertune-general-nlp-lora-tinyllama-1.1b-chat-v1.0")This is a LoRA adapter for TinyLlama/TinyLlama-1.1B-Chat-v1.0 fine-tuned with Flower federated learning framework on a general NLP dataset.
This model is a LoRA adapter fine-tuned on TinyLlama/TinyLlama-1.1B-Chat-v1.0 using the Flower federated learning framework. It was trained on a general NLP dataset (vicgalle/alpaca-gpt4) through distributed learning to improve performance.