jjuun commited on
Commit
1cf432c
1 Parent(s): 4efe2ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -13,6 +13,8 @@ def model_load():
13
  pipe = StableDiffusionXLPipeline.from_pretrained(
14
  "stabilityai/stable-diffusion-xl-base-1.0", vae=vae, torch_dtype=torch.float16
15
  )
 
 
16
  # load lora weight
17
  pipe.load_lora_weights("jjuun/vivid_color_style")
18
 
@@ -66,7 +68,8 @@ def create_demo():
66
 
67
  examples= [["a colorful fox", "20", "9", "0", "", "", "examples/fox.png"],
68
  ["a colorful messi", "20", "9", "191251724", "", "", "examples/messi.png"],
69
- ["a colorful pyramid", "20", "9", "0", "", "", "examples/pyramid.png"]],
 
70
  inputs = [prompt, num_steps, guidance_scale, seed, a_prompt, n_prompt, result]
71
  )
72
 
 
13
  pipe = StableDiffusionXLPipeline.from_pretrained(
14
  "stabilityai/stable-diffusion-xl-base-1.0", vae=vae, torch_dtype=torch.float16
15
  )
16
+ pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
17
+
18
  # load lora weight
19
  pipe.load_lora_weights("jjuun/vivid_color_style")
20
 
 
68
 
69
  examples= [["a colorful fox", "20", "9", "0", "", "", "examples/fox.png"],
70
  ["a colorful messi", "20", "9", "191251724", "", "", "examples/messi.png"],
71
+ ["a colorful pyramid", "20", "9", "0", "", "", "examples/pyramid.png"],
72
+ ["a colorful octopus playing violin", "20", "9", "0", "", "", "examples/octopus.png"]],
73
  inputs = [prompt, num_steps, guidance_scale, seed, a_prompt, n_prompt, result]
74
  )
75