1import sdkit
2from sdkit.generate import generate_images
3from sdkit.models import load_model
4from sdkit.utils import log, save_images
5import time
6import random
7from PIL import Image
8import PIL
9class Model():
10 def __init__(self):
11 self.context = sdkit.Context()
12 self.context.model_paths["stable-diffusion"] = "Deliberate_v3.safetensors" //(or any other model, i recommend Deliberate 2 or 6, May 2024)
13 self.context.model_paths["lora"] = "delib_6_8_f_m.safetensors"
14 load_model(self.context, "stable-diffusion")
15 load_model(self.context, "lora")
16 def generate_image(self, prompt, guidance_scale, seed, width, height, lora_alpha):
17 images = generate_images(self.context, prompt=prompt, guidance_scale=guidance_scale, sampler_name="dpmpp_2m", seed=seed, width=width, height=height, lora_alpha=lora_alpha)
18 name = "image"+str(time.strftime('%X')).replace(':','-')+".png"
19 path = "./static/nn/" + name
20 im1 = images[0].save(path)
21 return path
22...
23prompt = ...
24etc
25...
26im = self.model.generate_image(prompt, seed, width, height, lora_alpha)
27m = Model
prompt needs a race name, eg "dwarf female red hair"
2 > guidance_scale > 7
seed any
512 >= width >= 756
0.3 >= lora_alpha >= 0.7