Views
No views yet
tarot_stylepip install diffusers --upgradepip install invisible_watermark transformers accelerate safetensors1import torch
2from diffusers import DiffusionPipeline, AutoencoderKL
3
4vae = AutoencoderKL.from_pretrained('madebyollin/sdxl-vae-fp16-fix', torch_dtype=torch.float16)
5
6pipe = DiffusionPipeline.from_pretrained(
7 "stabilityai/stable-diffusion-xl-base-1.0",
8 vae=vae, torch_dtype=torch.float16, variant="fp16",
9 use_safetensors=True
10)
11
12# This is where you load your trained weights
13pipe.load_lora_weights('jtlowell/tarot')
14
15pipe.to("cuda")
16
17prompt = "A majestic tarot_style jumping from a big stone at night"
18
19image = pipe(prompt=prompt, num_inference_steps=50).images[0]