ford442 commited on
Commit
01d6e07
·
verified ·
1 Parent(s): 95d52b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -13,7 +13,7 @@ import gradio as gr
13
  import numpy as np
14
  from PIL import Image
15
  import torch
16
- from diffusers import AutoencoderKL, StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler, EDMDPMSolverMultistepScheduler
17
  from typing import Tuple
18
  import paramiko
19
  import gc
@@ -151,8 +151,8 @@ def load_and_prepare_model():
151
  #'ford442/RealVisXL_V5.0_BF16',
152
  #'ford442/Juggernaut-XI-v11-fp32',
153
  # 'SG161222/RealVisXL_V5.0',
154
- #'John6666/uber-realistic-porn-merge-xl-urpmxl-v3-sdxl',
155
- 'John6666/uber-realistic-porn-merge-xl-urpmxl-v6final-sdxl',
156
  #torch_dtype=torch.bfloat16,
157
  add_watermarker=False,
158
  # custom_pipeline="lpw_stable_diffusion_xl",
@@ -231,11 +231,10 @@ def load_and_prepare_model():
231
  #pipe.to(device, torch.bfloat16)
232
  #del pipeX
233
 
234
- #pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/Fantasy_World_XL.safetensors", adapter_name="fantasy")
235
  pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/skin_texture_style_v4.safetensors", adapter_name="skin")
236
  pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/epicNewPhoto.safetensors", adapter_name="photo")
237
- #pipe.set_adapters(["skin", "photo", "fantasy"], adapter_weights=[0.5, 0.5, 0.5])
238
- pipe.set_adapters(["skin", "photo"], adapter_weights=[0.80, 0.80])
239
 
240
  #sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", algorithm_type="dpmsolver++")
241
  #sched = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, beta_schedule="linear", algorithm_type="dpmsolver++")
 
13
  import numpy as np
14
  from PIL import Image
15
  import torch
16
+ from diffusers import AutoencoderKL, StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
17
  from typing import Tuple
18
  import paramiko
19
  import gc
 
151
  #'ford442/RealVisXL_V5.0_BF16',
152
  #'ford442/Juggernaut-XI-v11-fp32',
153
  # 'SG161222/RealVisXL_V5.0',
154
+ 'John6666/uber-realistic-porn-merge-xl-urpmxl-v3-sdxl',
155
+ #'John6666/uber-realistic-porn-merge-xl-urpmxl-v6final-sdxl',
156
  #torch_dtype=torch.bfloat16,
157
  add_watermarker=False,
158
  # custom_pipeline="lpw_stable_diffusion_xl",
 
231
  #pipe.to(device, torch.bfloat16)
232
  #del pipeX
233
 
234
+ pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/Fantasy_World_XL.safetensors", adapter_name="fantasy")
235
  pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/skin_texture_style_v4.safetensors", adapter_name="skin")
236
  pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/epicNewPhoto.safetensors", adapter_name="photo")
237
+ pipe.set_adapters(["skin", "photo", "fantasy"], adapter_weights=[0.75, 0.25, 0.5])
 
238
 
239
  #sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", algorithm_type="dpmsolver++")
240
  #sched = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, beta_schedule="linear", algorithm_type="dpmsolver++")