Update app.py
Browse files
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 |
|