Gustavosta/Stable-Diffusion-Prompts
Viewer • Updated • 81.9k • 12.6k • 523
How to use Technotech/MagicPrompt-tinystories-33M-epoch2 with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("models/TinyStories-33M")
model = PeftModel.from_pretrained(base_model, "Technotech/MagicPrompt-tinystories-33M-epoch2")Rank 8 LoRA, 2 epochs of Stable-Diffusion-Prompts with batch size 128.
The following bitsandbytes quantization config was used during training: