Staticaliza commited on
Commit
13c2b13
·
verified ·
1 Parent(s): 3439ef5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -44,16 +44,16 @@ repo_default.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
44
  repo_default.set_adapters("base")
45
  # repo_default.set_adapters(["base"], adapter_weights=[0.7])
46
 
47
- repo_pixel = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
48
- repo_pixel.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="base")
49
- repo_pixel.load_lora_weights("nerijs/pixel-art-xl", adapter_name="base2")
50
- repo_pixel.set_adapters(["base", "base2"], adapter_weights=[1, 1])
51
 
52
  repo_customs = {
53
  "Default": repo_default,
54
- "Realistic": StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V4.0", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
55
- "Anime": StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
56
- "Pixel": repo_pixel,
57
  }
58
 
59
  print("-----------A")
@@ -61,7 +61,7 @@ print(repo_default.scheduler)
61
  print("-----/-////B")
62
  print(EulerAncestralDiscreteScheduler.from_config(repo_default.scheduler.config))
63
  print("-----------Z")
64
- repo_default.scheduler = EulerAncestralDiscreteScheduler.from_config(repo_default.scheduler.config)
65
 
66
 
67
  # Functions
 
44
  repo_default.set_adapters("base")
45
  # repo_default.set_adapters(["base"], adapter_weights=[0.7])
46
 
47
+ # repo_pixel = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
48
+ # repo_pixel.load_lora_weights("artificialguybr/PixelArtRedmond", adapter_name="base")
49
+ # repo_pixel.load_lora_weights("nerijs/pixel-art-xl", adapter_name="base2")
50
+ # repo_pixel.set_adapters(["base", "base2"], adapter_weights=[1, 1])
51
 
52
  repo_customs = {
53
  "Default": repo_default,
54
+ "Realistic": None, # StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V4.0", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
55
+ "Anime": None, # StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False),
56
+ "Pixel": None, # repo_pixel,
57
  }
58
 
59
  print("-----------A")
 
61
  print("-----/-////B")
62
  print(EulerAncestralDiscreteScheduler.from_config(repo_default.scheduler.config))
63
  print("-----------Z")
64
+ # repo_default.scheduler = EulerAncestralDiscreteScheduler.from_config(repo_default.scheduler.config)
65
 
66
 
67
  # Functions