Staticaliza commited on
Commit
7a28cde
·
verified ·
1 Parent(s): c9612cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -40,18 +40,18 @@ controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-1
40
 
41
  repo_default = StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-Final", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
42
 
43
- repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
44
- repo_default.set_adapters("base")
45
 
46
- repo_pixel = StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-Final", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
47
- repo_pixel.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="base")
48
- repo_pixel.load_lora_weights("nerijs/pixel-art-xl", adapter_name="base2")
49
- repo_pixel.set_adapters(["base", "base2"], adapter_weights=[1, 1])
50
 
51
  repo_customs = {
52
  "Default": repo_default,
53
- "Realistic": StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V4.0", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
54
- "Anime": StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
55
  "Pixel": repo_pixel,
56
  }
57
 
 
40
 
41
  repo_default = StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-Final", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
42
 
43
+ #repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
44
+ #repo_default.set_adapters("base")
45
 
46
+ repo_pixel = None#StableDiffusionXLPipeline.from_pretrained("fluently/Fluently-XL-Final", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
47
+ #repo_pixel.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="base")
48
+ #repo_pixel.load_lora_weights("nerijs/pixel-art-xl", adapter_name="base2")
49
+ #repo_pixel.set_adapters(["base", "base2"], adapter_weights=[1, 1])
50
 
51
  repo_customs = {
52
  "Default": repo_default,
53
+ "Realistic": None,#StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V4.0", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
54
+ "Anime": None,#StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
55
  "Pixel": repo_pixel,
56
  }
57