Ir para o conteúdo
ModelVerificadoExterno
Baixar

Flower LoRA Weights

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

TipoLoRAModelo baserunwayml/stable-diffusion-v1-5Versãoadapter_modelPublicador@Pritish23Cnota

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

Fontes

1 fonte
Hugging FacePrimárioVersão c0f1491cc12bde0ae081ce956106c00b10474a4c
huggingface.co/Pritish23/flower-lora-weights/tree/c0f1491cc12bde0ae081ce956106c00b10474a4c

Verificado 3 de out.

Artefatos de modelo

1 artefato

adapter_model.safetensors

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

Baixar

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