Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
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 |
-
|
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 |
-
|
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 |
-
|
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++")
|