|
|
--- |
|
|
license: apache-2.0 |
|
|
base_model: stabilityai/stable-diffusion-xl-base-1.0 |
|
|
datasets: |
|
|
- Oysiyl/google-android-toy-sdxl |
|
|
language: |
|
|
- en |
|
|
--- |
|
|
|
|
|
|
|
|
### Model card |
|
|
|
|
|
Finetuned from SDXL using LoRA. |
|
|
|
|
|
W&B [run](https://wandb.ai/logart1995/text2image-fine-tune/runs/3tza87bz?workspace=user-logart1995). |
|
|
|
|
|
### Inference |
|
|
|
|
|
```py |
|
|
from diffusers import AutoPipelineForText2Image |
|
|
import torch |
|
|
|
|
|
pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16) |
|
|
pipe.load_lora_weights("Oysiyl/sdxl-lora-android-google-toy", weights="pytorch_lora_weights.safetensors") |
|
|
pipe = pipe.to("cuda") |
|
|
|
|
|
g = torch.Generator(device="cuda").manual_seed(42) |
|
|
|
|
|
image = pipe("An android toy near Eiffel tower", |
|
|
num_inference_steps=50, |
|
|
num_images_per_prompt=1, |
|
|
guidance_scale=7.5, |
|
|
temperature=1.0, |
|
|
generator=g).images[0] |
|
|
|
|
|
image.save("android_toy_sdxl.png") |
|
|
``` |
|
|
|
|
|
### Output |
|
|
|
|
|
 |
