Krebzonide commited on
Commit
1780330
·
1 Parent(s): d3c021a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,13 +9,13 @@ import gradio as gr
9
  model_base = "Krebzonide/LazyMixPlus" #nsfw people
10
  #model_base = "Y2G/humans-v1" #boring people
11
 
12
- lora_model_path = "Krebzonide/LoRA-CH-0" #mecjh - Corey H, traind on epiCRealism
13
  #lora_model_path = "Krebzonide/LoRA-CH-1" #mecjh - Corey H, traind on epiCRealism
14
- #lora_model_path = "Krebzonide/LoRA-EM1" #exgfem - Emily M, trained on LizyMixPlus
15
  #lora_model_path = "Krebzonide/LoRA-YX1" #uwspyx - Professor Xing, trained on Realistic_Vision
16
 
17
  pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16, use_safetensors=True, use_auth_token="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")
18
- pipe.unet.load_attn_procs(lora_model_path)
19
  pipe.to("cuda")
20
 
21
  css = """
 
9
  model_base = "Krebzonide/LazyMixPlus" #nsfw people
10
  #model_base = "Y2G/humans-v1" #boring people
11
 
12
+ #lora_model_path = "Krebzonide/LoRA-CH-0" #mecjh - Corey H, traind on epiCRealism
13
  #lora_model_path = "Krebzonide/LoRA-CH-1" #mecjh - Corey H, traind on epiCRealism
14
+ lora_model_path = "Krebzonide/LoRA-EM1" #exgfem - Emily M, trained on LizyMixPlus
15
  #lora_model_path = "Krebzonide/LoRA-YX1" #uwspyx - Professor Xing, trained on Realistic_Vision
16
 
17
  pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16, use_safetensors=True, use_auth_token="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")
18
+ pipe.unet.load_attn_procs(lora_model_path, use_auth_token="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")
19
  pipe.to("cuda")
20
 
21
  css = """