Update app.py
Browse files
app.py
CHANGED
|
@@ -44,8 +44,8 @@ repo_pixel.load_lora_weights("nerijs/pixel-art-xl", adapter_name="base2")
|
|
| 44 |
repo_pixel.set_adapters(["base", "base2"], adapter_weights=[1, 1])
|
| 45 |
|
| 46 |
repo_large = StableDiffusionXLPipeline.from_pretrained("Corcelio/mobius", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
| 47 |
-
|
| 48 |
-
|
| 49 |
|
| 50 |
repo_customs = {
|
| 51 |
"Default": repo_default,
|
|
@@ -90,8 +90,8 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
|
|
| 90 |
steps_set = 8
|
| 91 |
guidance_set = 1.5
|
| 92 |
elif model == "Large":
|
| 93 |
-
steps_set =
|
| 94 |
-
guidance_set =
|
| 95 |
else:
|
| 96 |
steps_set = 16
|
| 97 |
guidance_set = 3
|
|
|
|
| 44 |
repo_pixel.set_adapters(["base", "base2"], adapter_weights=[1, 1])
|
| 45 |
|
| 46 |
repo_large = StableDiffusionXLPipeline.from_pretrained("Corcelio/mobius", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
| 47 |
+
repo_large.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
| 48 |
+
repo_large.set_adapters(["base"], adapter_weights=[0.5])
|
| 49 |
|
| 50 |
repo_customs = {
|
| 51 |
"Default": repo_default,
|
|
|
|
| 90 |
steps_set = 8
|
| 91 |
guidance_set = 1.5
|
| 92 |
elif model == "Large":
|
| 93 |
+
steps_set = 16
|
| 94 |
+
guidance_set = 7
|
| 95 |
else:
|
| 96 |
steps_set = 16
|
| 97 |
guidance_set = 3
|