Krebzonide commited on
Commit
90c02c2
·
1 Parent(s): 9e285f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -2,12 +2,15 @@ import torch
2
  from diffusers import StableDiffusionPipeline
3
  import gradio as gr
4
 
 
 
 
5
  #model_base = "runwayml/stable-diffusion-v1-5" #base
6
- #model_base = "SG161222/Realistic_Vision_V5.1_noVAE" #people
7
- model_base = "Krebzonide/LazyMixPlus" #nsfw people
8
 
9
- #lora_model_path = "Krebzonide/LoRA-EM1" #Emily Maguire LoRA, trained on SD V1-5
10
- lora_model_path = "Krebzonide/LoRA-YX1" #Professor Xing LoRA, trained on Realistic_Vision
 
11
 
12
  pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16, use_safetensors=True)
13
  pipe.unet.load_attn_procs(lora_model_path) #working, commented to test stuff------------------------------------------
 
2
  from diffusers import StableDiffusionPipeline
3
  import gradio as gr
4
 
5
+ #model_base = "SG161222/Realistic_Vision_V5.1_noVAE" #realistic people
6
+ model_base = "emilianJR/epiCRealism" #realistic people
7
+ #model_base = "Lykon/DreamShaper" #unrealistic people
8
  #model_base = "runwayml/stable-diffusion-v1-5" #base
9
+ #model_base = "Krebzonide/LazyMixPlus" #nsfw people
 
10
 
11
+ lora_model_path = "Krebzonide/LoRA-CH-0" #Corey H, traind on epiCRealism
12
+ #lora_model_path = "Krebzonide/LoRA-EM1" #Emily M, trained on sd-V1-5
13
+ #lora_model_path = "Krebzonide/LoRA-YX1" #Professor Xing, trained on Realistic_Vision
14
 
15
  pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16, use_safetensors=True)
16
  pipe.unet.load_attn_procs(lora_model_path) #working, commented to test stuff------------------------------------------