Commit
·
bacf257
1
Parent(s):
468aab2
Update app.py
Browse files
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 |
-
|
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=
|
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 |
|