text_to_WebPage / README.md
keyurhirpara's picture
Update README.md
785eff0 verified
metadata
base_model: stabilityai/stable-diffusion-xl-base-1.0
library_name: diffusers
license: creativeml-openrail-m
tags:
  - stable-diffusion-xl
  - stable-diffusion-xl-diffusers
  - text-to-image
  - diffusers
  - diffusers-training
  - lora
inference: true
datasets:
  - keyurhirpara/Website_Img_5
language:
  - en
pipeline_tag: text-to-image

text2webpage fine-tuning - keyurhirpara/Text_to_WebPage

These are LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0. The weights were fine-tuned on the keyurhirpara/Website_Img_5 dataset. You can find some example images in the following.

img_0 img_1 img_2 img_3

Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.

Intended uses & limitations

How to use

from diffusers import DiffusionPipeline
import torch

model_path = "keyurhirpara/text_to_WebPage"
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True,)
pipe.to("cuda")
pipe.load_lora_weights(model_path)

prompt = "Create a English website page for Travel Agency: A visually immersive design with high-resolution images, a center-aligned logo, and a top navigation menu with links for vacation packages, destinations, and customer reviews. The design includes scrolling sections for showcasing destinations and highlighting the latest discount offers."

images = pipe(prompt=prompt, num_inference_steps=50, guidance_scale=7.5).images[0]

Limitations and bias

[TODO: provide examples of latent issues and potential remediations]

Training details

[TODO: describe the data used to train the model]