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 |
|