Aller au contenu
ModelVérifiéExterne
Télécharger

Flower LoRA Weights

LoRA fine-tuned weights for Stable Diffusion v1.5 trained on the Oxford-102 Flowers dataset.

Sourcehuggingface.co/Pritish23/flower-lora-weights/tree/c0f1491cc12bde0ae081ce956106c00b10474a4c
TypeLoRAModèle de baserunwayml/stable-diffusion-v1-5Versionadapter_modelÉditeur@Pritish23CClasse

Source du modèle

Type
LoRA
Taille du fichier
12,2 MB
Modèle de base
runwayml/stable-diffusion-v1-5
Version
vadapter_model
Mis à jour
3 oct. 2026
Éditeur
@Pritish23
Téléchargements
9
Tâches
Texte en image
Source
Hugging Face

Description de la source

🌸 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

Sources

1 source
Hugging FacePrincipalVersion c0f1491cc12bde0ae081ce956106c00b10474a4c
huggingface.co/Pritish23/flower-lora-weights/tree/c0f1491cc12bde0ae081ce956106c00b10474a4c

Vérifié 3 oct.

Artefacts du modèle

1 artefact

adapter_model.safetensors

safetensors · 12,2 MB · SHA-256 065314bcb3cd…78e9 · Hugging Face

Télécharger

Extraits de sources

3 extraits
Readme11 juil.huggingface.co/Pritish23/flower-lora-weights

--- 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)

Note3 oct.huggingface.co/Pritish23/flower-lora-weights

Source context: 9 downloads · 1 likes · Pipeline text-to-image · Library peft · Repo Pritish23/flower-lora-weights

Note17 juil.huggingface.co/Pritish23/flower-lora-weights

Source context: 13 downloads · 1 likes · Pipeline text-to-image · Library peft · Repo Pritish23/flower-lora-weights