Views
No views yet


1from diffusers import StableDiffusionInpaintPipeline
2pipe = StableDiffusionInpaintPipeline.from_pretrained(
3 args.pretrained_model_name_or_path,
4 torch_dtype=torch.float16
5).to('cuda')
6pipe.load_lora_weights("out/pytorch_lora_weights.safetensors")
7
8def generate(pipeline, prompt,image,mask,seed):
9 generator = torch.Generator(device='cuda').manual_seed(seed)
10 return pipeline(prompt=prompt,image=image,mask_image=mask,num_inference_steps=50, generator=generator).images[0]
11prmot=' black and white, patterned tank top with a relaxed fit, short sleeves, and an untucked style. The blouse is hip-length, made of cotton, and has a crew neckline.'
12generate(pipeline=pipe,prompt=prmot,image=sample['image'],mask=sample['mask'],seed=42)
13