Views
No views yet
1from diffusers import AutoPipelineForText2Image
2import torch
3
4pipeline = AutoPipelineForText2Image.from_pretrained('runwayml/stable-diffusion-v1-5', torch_dtype=torch.float16).to('cuda')
5pipeline.load_lora_weights('brushpenbob/wizard-of-oz-1-0', weight_name='wizerd_of_oz.safetensors')
6image = pipeline('Your custom prompt').images[0]