Update app.py
Browse files
app.py
CHANGED
@@ -38,11 +38,16 @@ repo_default = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flas
|
|
38 |
repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
39 |
repo_default.set_adapters(["base"], adapter_weights=[0.7])
|
40 |
|
|
|
|
|
|
|
|
|
|
|
41 |
repo_customs = {
|
42 |
"Default": repo_default,
|
43 |
"Realistic": StableDiffusionXLPipeline.from_pretrained("stablediffusionapi/NightVision_XL", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=False, add_watermarker=False),
|
44 |
"Anime": StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
|
45 |
-
"Pixel":
|
46 |
}
|
47 |
|
48 |
# Functions
|
|
|
38 |
repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
39 |
repo_default.set_adapters(["base"], adapter_weights=[0.7])
|
40 |
|
41 |
+
repo_pixel = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
42 |
+
repo_pixel.load_lora_weights("latent-consistency/lcm-lora-sdxl", adapter_name="lora")
|
43 |
+
repo_pixel.load_lora_weights("nerijs/pixel-art-xl", adapter_name="pixel")
|
44 |
+
repo_pixel.set_adapters(["lora", "pixel"], adapter_weights=[1.0, 1.2])
|
45 |
+
|
46 |
repo_customs = {
|
47 |
"Default": repo_default,
|
48 |
"Realistic": StableDiffusionXLPipeline.from_pretrained("stablediffusionapi/NightVision_XL", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=False, add_watermarker=False),
|
49 |
"Anime": StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
|
50 |
+
"Pixel": repo_pixel,
|
51 |
}
|
52 |
|
53 |
# Functions
|