Flower LoRA Weights
LoRA fine-tuned weights for Stable Diffusion v1.5 trained on the Oxford-102 Flowers dataset.
Fonte do modelo
- Tipo
- LoRA
- Tamanho do arquivo
- 12,2 MB
- Modelo base
- runwayml/stable-diffusion-v1-5
- Versão
- vadapter_model
- Atualizado
- 3 de out. de 2026
- Publicador
- @Pritish23
- Downloads
- 9
- Tarefas
- Texto para imagem
- Fonte
- Hugging Face
Descrição da fonte
🌸 Flower LoRA Weights
LoRA fine-tuned weights for Stable Diffusion v1.5 trained on the Oxford-102 Flowers dataset.
Model Details
- Base model: Stable Diffusion v1.5
- Fine-tuning: LoRA (r=16, alpha=32)
- Dataset: Oxford-102 Flowers (1,020 images)
- Epochs: 10
- CLIP Score: 0.315
Usage
from diffusers import StableDiffusionPipeline
from peft import PeftModel
import torch
pipeline = StableDiffusionPipeline.from_pretrained(
"runwayml/stable-diffusion-v1-5",
torch_dtype=torch.float16
)
pipeline.unet = PeftModel.from_pretrained(
pipeline.unet,
"Pritish23/flower-lora-weights"
)
pipeline.unet = pipeline.unet.merge_and_unload()
pipeline = pipeline.to("cuda")
image = pipeline("a photo of a sunflower, vibrant colors").images[0]
image.save("sunflower.png")
Training Details
- Target modules: to_q, to_k, to_v, to_out
- Learning rate: 1e-4
- Batch size: 1
- Optimizer: AdamW
Demo
Try the live demo: Text-to-Image Generator Space
Author
Pritish Sharma — B.Sc. Data Science
- GitHub: Pritish-23
- LinkedIn: pritishsharma230805
Fontes
1 fonte
Fontes
1 fonteVerificado 3 de out.
Artefatos de modelo
1 artefato
Artefatos de modelo
1 artefatoTrechos de fonte
3 trechos
Trechos de fonte
3 trechos--- base_model: runwayml/stable-diffusion-v1-5 library_name: peft tags: - stable-diffusion - lora - text-to-image - flowers license: mit --- # 🌸 Flower LoRA Weights LoRA fine-tuned weights for Stable Diffusion v1.5 trained on the Oxford-102 Flowers dataset. ## Model Details - **Base model:** Stable Diffusion v1.5 - **Fine-tuning:** LoRA (r=16, alpha=32) - **Dataset:** Oxford-102 Flowers (1,020 images) - **Epochs:** 10 - **CLIP Score:** 0.315 ## Usage ```python from diffusers import StableDiffusionPipeline from peft import PeftModel import torch pipeline = StableDiffusionPipeline.from_pretrained( "runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16 ) pipeline.unet = PeftModel.from_pretrained( pipeline.unet, "Pritish23/flower-lora-weights" ) pipeline.unet = pipeline.unet.merge_and_unload() pipeline = pipeline.to("cuda") image = pipeline("a photo of a sunflower, vibrant colors").images[0] image.save("sunflower.png") ``` ## Training Details - Target modules: to_q, to_k, to_v, to_out - Learning rate: 1e-4 - Batch size: 1 - Optimizer: AdamW ## Demo Try the live demo: [Text-to-Image Generator Space](https://huggingface.co/spaces/Pritish23/text-to-image-generator) ## Author **Pritish Sharma** — B.Sc. Data Science - GitHub: [Pritish-23](https://github.com/Pritish-23) - LinkedIn: [pritishsharma230805](https://linkedin.com/in/pritishsharma230805)
Source context: 9 downloads · 1 likes · Pipeline text-to-image · Library peft · Repo Pritish23/flower-lora-weights
Source context: 13 downloads · 1 likes · Pipeline text-to-image · Library peft · Repo Pritish23/flower-lora-weights
