hsuwill000 commited on
Commit
5990b17
1 Parent(s): c8e5594

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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,