Views
No views yet


pip install torch diffusers accelerate pillow1from diffusers import StableDiffusionPipeline
2import torch
3from PIL import Image
4
5def image_grid(imgs, rows, cols):
6 assert len(imgs) == rows*cols
7
8 w, h = imgs[0].size
9 grid = Image.new('RGB', size=(cols*w, rows*h))
10 grid_w, grid_h = grid.size
11
12 for i, img in enumerate(imgs):
13 grid.paste(img, box=(i%cols*w, i//cols*h))
14 return grid
15
16model_id = "CHAITron/KranokThaiV1"
17pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
18pipe = pipe.to("cuda")
19
20prompt = "a colourful masterpiece thai dog in <knt-style> style, high-quality, professional art, realistic, thai style"
21
22num_samples = 4
23num_rows = 1
24
25all_images = []
26for _ in range(num_rows):
27 images = pipe(prompt, num_images_per_prompt=num_samples, num_inference_steps=25, guidance_scale=9).images
28 all_images.extend(images)
29
30grid = image_grid(all_images, num_rows, num_samples)
31grid