Krebzonide commited on
Commit
f01e148
·
1 Parent(s): 2a73eea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -2,8 +2,12 @@ import torch
2
  from diffusers import StableDiffusionPipeline
3
  import gradio as gr
4
 
5
- model_base = "Krebzonide/LazyMixPlus"
6
- lora_model_path = "Krebzonide/94g1-jemz-41r2-0"
 
 
 
 
7
 
8
  pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16, use_safetensors=True)
9
  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 = "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------------------------------------------