Krebzonide commited on
Commit
bacf257
·
1 Parent(s): 468aab2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,8 +9,8 @@ import gradio as gr
9
  model_base = "Krebzonide/LazyMixPlus" #nsfw people
10
 
11
  #lora_model_path = "Krebzonide/LoRA-CH-0" #mecjh - Corey H, traind on epiCRealism
12
- lora_model_path = "Krebzonide/LoRA-CH-1" #mecjh - Corey H, traind on epiCRealism
13
- #lora_model_path = "Krebzonide/LoRA-EM1" #exgfem - Emily M, trained on LizyMixPlus
14
  #lora_model_path = "Krebzonide/LoRA-YX1" #uwspyx - Professor Xing, trained on Realistic_Vision
15
 
16
  pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16, use_safetensors=True)
@@ -35,7 +35,7 @@ def generate(prompt, neg_prompt, samp_steps, guide_scale, lora_scale, progress=g
35
  num_inference_steps=samp_steps,
36
  guidance_scale=guide_scale,
37
  cross_attention_kwargs={"scale": lora_scale},
38
- num_images_per_prompt=4
39
  ).images
40
  return [(img, f"Image {i+1}") for i, img in enumerate(images)]
41
 
 
9
  model_base = "Krebzonide/LazyMixPlus" #nsfw people
10
 
11
  #lora_model_path = "Krebzonide/LoRA-CH-0" #mecjh - Corey H, traind on epiCRealism
12
+ #lora_model_path = "Krebzonide/LoRA-CH-1" #mecjh - Corey H, traind on epiCRealism
13
+ lora_model_path = "Krebzonide/LoRA-EM1" #exgfem - Emily M, trained on LizyMixPlus
14
  #lora_model_path = "Krebzonide/LoRA-YX1" #uwspyx - Professor Xing, trained on Realistic_Vision
15
 
16
  pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16, use_safetensors=True)
 
35
  num_inference_steps=samp_steps,
36
  guidance_scale=guide_scale,
37
  cross_attention_kwargs={"scale": lora_scale},
38
+ num_images_per_prompt=6
39
  ).images
40
  return [(img, f"Image {i+1}") for i, img in enumerate(images)]
41