Staticaliza commited on
Commit
32f63e2
·
verified ·
1 Parent(s): c00ea85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
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": StableDiffusionXLPipeline.from_pretrained("nerijs/pixel-art-xl", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
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