Update app.py
Browse files
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
|
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",
|