Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,7 +28,6 @@ pipe = OVStableDiffusionPipeline.from_pretrained(
|
|
| 28 |
#torch_dtype=torch.IntTensor, #慢,
|
| 29 |
safety_checker=None,
|
| 30 |
use_safetensors=False,
|
| 31 |
-
scheduler=LCMScheduler,
|
| 32 |
)
|
| 33 |
|
| 34 |
|
|
@@ -54,6 +53,7 @@ def infer(prompt,negative_prompt):
|
|
| 54 |
negative_prompt = negative_prompt,
|
| 55 |
width = WIDTH,
|
| 56 |
height = HIGH,
|
|
|
|
| 57 |
guidance_scale=1.0,
|
| 58 |
num_inference_steps=8,
|
| 59 |
num_images_per_prompt=1,
|
|
|
|
| 28 |
#torch_dtype=torch.IntTensor, #慢,
|
| 29 |
safety_checker=None,
|
| 30 |
use_safetensors=False,
|
|
|
|
| 31 |
)
|
| 32 |
|
| 33 |
|
|
|
|
| 53 |
negative_prompt = negative_prompt,
|
| 54 |
width = WIDTH,
|
| 55 |
height = HIGH,
|
| 56 |
+
scheduler=LCMScheduler,
|
| 57 |
guidance_scale=1.0,
|
| 58 |
num_inference_steps=8,
|
| 59 |
num_images_per_prompt=1,
|