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('your-username/your-lora-model-name', weight_name='pytorch_lora_weights.safetensors')
6
7image = pipeline('your prompt here').images[0]
8image.save('output.png')