hsuwill000 commited on
Commit
c6b6ad5
1 Parent(s): 81ed687

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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 = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
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=7.5,
58
- num_inference_steps=26,
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