Views
No views yet
1from diffusers import StableDiffusionXLPipeline
2import torch
3
4pipeline = StableDiffusionXLPipeline.from_pretrained(
5 "stabilityai/stable-diffusion-xl-base-1.0",
6 torch_dtype=torch.float16,
7 variant="fp16",
8 use_safetensors=True
9).to("cuda")
10
11pipeline.load_lora_weights("MohamedAhmedAE/stable-diffusion-xl-base_lora_finetuning")
12
13prompt = "a cute pokemon character"
14image = pipeline(prompt, num_inference_steps=20, guidance_scale=7.5).images[0]
15image.save("generated_image.png")1{
2 "base_model": "stabilityai/stable-diffusion-xl-base-1.0",
3 "epochs": 30,
4 "batch_size": 1,
5 "learning_rate": 5e-05,
6 "lora_rank": 4,
7 "lora_alpha": 32,
8 "resolution": 1024
9}