Disty0 commited on
Commit
a390436
1 Parent(s): e3761d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -28,7 +28,7 @@ Converted from [SoteMix](https://huggingface.co/Disty0/SoteMix) to [LCM_SoteMix]
28
 
29
  [LCM Project page](https://latent-consistency-models.github.io)
30
 
31
- <p>Running on CPU with OpenVINO.</p>
32
  '''
33
 
34
  MAX_SEED = np.iinfo(np.int32).max
@@ -39,7 +39,6 @@ batch_size = -1
39
  width = int(os.getenv("IMAGE_WIDTH", "512"))
40
  height = int(os.getenv("IMAGE_HEIGHT", "512"))
41
  num_images = int(os.getenv("NUM_IMAGES", "1"))
42
- guidance_scale = float(os.getenv("GUIDANCE_SCALE", "1.0"))
43
 
44
  class CustomOVModelVaeDecoder(OVModelVaeDecoder):
45
  def __init__(
@@ -142,6 +141,13 @@ with gr.Blocks(css="style.css") as demo:
142
  randomize=True
143
  )
144
  randomize_seed = gr.Checkbox(label="Randomize seed across runs", value=True)
 
 
 
 
 
 
 
145
  with gr.Row():
146
  num_inference_steps = gr.Slider(
147
  label="Number of inference steps for base",
 
28
 
29
  [LCM Project page](https://latent-consistency-models.github.io)
30
 
31
+ <p>Running on CPU with OpenVINO acceleration</p>
32
  '''
33
 
34
  MAX_SEED = np.iinfo(np.int32).max
 
39
  width = int(os.getenv("IMAGE_WIDTH", "512"))
40
  height = int(os.getenv("IMAGE_HEIGHT", "512"))
41
  num_images = int(os.getenv("NUM_IMAGES", "1"))
 
42
 
43
  class CustomOVModelVaeDecoder(OVModelVaeDecoder):
44
  def __init__(
 
141
  randomize=True
142
  )
143
  randomize_seed = gr.Checkbox(label="Randomize seed across runs", value=True)
144
+ guidance_scale = gr.Slider(
145
+ label="Guidance scale for base",
146
+ minimum=1,
147
+ maximum=8,
148
+ step=0.1,
149
+ value=1.0,
150
+ )
151
  with gr.Row():
152
  num_inference_steps = gr.Slider(
153
  label="Number of inference steps for base",