Views
No views yet
1from diffusers import StableDiffusionPipeline
2import torch
3import os
4from PIL import Image
5model_id = "michaelz/MakotoShinkai-Style-Diffusion"
6pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
7pipe.requires_safety_checker = False
8pipe.safety_checker = None
9pipe = pipe.to("cuda")
10
11save_dir = os.path.join("results")
12os.makedirs(save_dir,exist_ok=True)
13prompt_list = [
14 "MakotoShinkaiStyle, a beautiful sky",
15 "MakotoShinkaiStyle, a beautiful landscape",
16 "MakotoShinkaiStyle, a beautiful city"]
17for prompt in prompt_list:
18 torch.manual_seed(-1)
19 image = pipe(prompt,width = 512,height = 512).images[0]
20 save_path = os.path.join(save_dir,prompt + ".png")
21 image.save(save_path)


