hsuwill000 commited on
Commit
82e6660
·
verified ·
1 Parent(s): bc4e9b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -42,7 +42,7 @@ pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{tae
42
  """
43
  print(pipe.scheduler.compatibles)
44
 
45
- pipe.reshape(batch_size=batch_size, height=HIGH, width=WIDTH, num_images_per_prompt=1)
46
 
47
  pipe.compile()
48
 
@@ -55,9 +55,10 @@ def infer(prompt, negative_prompt, num_inference_steps=1):
55
  negative_prompt=negative_prompt,
56
  width=WIDTH,
57
  height=HIGH,
58
- guidance_scale=1.0,
59
  num_inference_steps=num_inference_steps,
60
  num_images_per_prompt=1,
 
61
  ).images[0]
62
 
63
  return image
 
42
  """
43
  print(pipe.scheduler.compatibles)
44
 
45
+ pipe.reshape(batch_size=batch_size, height=HIGH, width=WIDTH, num_images_per_prompt=2)
46
 
47
  pipe.compile()
48
 
 
55
  negative_prompt=negative_prompt,
56
  width=WIDTH,
57
  height=HIGH,
58
+ guidance_scale=0.0,
59
  num_inference_steps=num_inference_steps,
60
  num_images_per_prompt=1,
61
+ strength=0.5,
62
  ).images[0]
63
 
64
  return image