Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ import torch
|
|
| 7 |
from huggingface_hub import snapshot_download
|
| 8 |
import openvino.runtime as ov
|
| 9 |
from typing import Optional, Dict
|
| 10 |
-
from diffusers import EulerDiscreteScheduler
|
| 11 |
|
| 12 |
|
| 13 |
model_id = "hsuwill000/anything-v5-openvino"
|
|
@@ -29,7 +29,7 @@ pipe = OVStableDiffusionPipeline.from_pretrained(
|
|
| 29 |
safety_checker=None,
|
| 30 |
use_safetensors=False,
|
| 31 |
)
|
| 32 |
-
pipe.scheduler =
|
| 33 |
|
| 34 |
|
| 35 |
|
|
@@ -54,8 +54,8 @@ def infer(prompt,negative_prompt):
|
|
| 54 |
negative_prompt = negative_prompt,
|
| 55 |
width = WIDTH,
|
| 56 |
height = HIGH,
|
| 57 |
-
guidance_scale=
|
| 58 |
-
num_inference_steps=
|
| 59 |
num_images_per_prompt=1,
|
| 60 |
).images[0]
|
| 61 |
|
|
|
|
| 7 |
from huggingface_hub import snapshot_download
|
| 8 |
import openvino.runtime as ov
|
| 9 |
from typing import Optional, Dict
|
| 10 |
+
from diffusers import EulerDiscreteScheduler,LCMScheduler
|
| 11 |
|
| 12 |
|
| 13 |
model_id = "hsuwill000/anything-v5-openvino"
|
|
|
|
| 29 |
safety_checker=None,
|
| 30 |
use_safetensors=False,
|
| 31 |
)
|
| 32 |
+
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
|
| 33 |
|
| 34 |
|
| 35 |
|
|
|
|
| 54 |
negative_prompt = negative_prompt,
|
| 55 |
width = WIDTH,
|
| 56 |
height = HIGH,
|
| 57 |
+
guidance_scale=1.0,
|
| 58 |
+
num_inference_steps=6,
|
| 59 |
num_images_per_prompt=1,
|
| 60 |
).images[0]
|
| 61 |
|