Views
No views yet
1# TODO: add an example code snippet for running this diffusion pipeline
2import torch
3from diffusers import StableDiffusionPipeline
4
5
6model_path = "iiiiDeal/SD1_5_Rank_8_CulText_10k"
7prompt = "A high resolution image of a modern security checkpoint in China, featuring bilingual signage, uniformed personnel, and sleek architectural design."
8pipe = StableDiffusionPipeline.from_pretrained("sd-legacy/stable-diffusion-v1-5").to("cuda")
9
10image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0]
11image.save("samples/pretrain.png")
12
13
14pipe.unet.load_attn_procs(model_path)
15
16image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0]
17image.save("samples/finetune.png")