Views
No views yet
1import torch
2from diffusers import DiffusionPipeline
3
4# Load Stable Diffusion XL Base1.0
5pipeline = DiffusionPipeline.from_pretrained(
6 "stabilityai/stable-diffusion-xl-base-1.0",
7 torch_dtype=torch.float16,
8 variant="fp16",
9 use_safetensors=True
10).to("cuda")
11
12# Optional CPU offloading to save some GPU Memory
13pipeline.enable_model_cpu_offload()
14
15# Loading Trained DreamBooth LoRA Weights
16pipeline.load_lora_weights("adhisetiawan/sdxl-base-1.0-indonesian-food-dreambooth-lora")
17
18images = pipeline(
19 "a delicous takoyaki in a plate", num_images_per_prompt=4, guidance_scale=8
20 )
21for i in range(len(images.images)):
22 display(images.images[i])