Update demo_web.py
Browse files- demo_web.py +1 -1
demo_web.py
CHANGED
@@ -104,7 +104,7 @@ def load_engine(args):
|
|
104 |
if __name__ == "__main__":
|
105 |
parser = argparse.ArgumentParser()
|
106 |
# pipeline configure
|
107 |
-
parser.add_argument("--model", type=str, default="
|
108 |
# scheduler params
|
109 |
parser.add_argument("--beta-start", type=float, default=0.00085, help="LMSDiscreteScheduler::beta_start")
|
110 |
parser.add_argument("--beta-end", type=float, default=0.012, help="LMSDiscreteScheduler::beta_end")
|
|
|
104 |
if __name__ == "__main__":
|
105 |
parser = argparse.ArgumentParser()
|
106 |
# pipeline configure
|
107 |
+
parser.add_argument("--model", type=str, default="4eJIoBek/stable-diffusion-v1-4-openvino-fp32", help="model name")
|
108 |
# scheduler params
|
109 |
parser.add_argument("--beta-start", type=float, default=0.00085, help="LMSDiscreteScheduler::beta_start")
|
110 |
parser.add_argument("--beta-end", type=float, default=0.012, help="LMSDiscreteScheduler::beta_end")
|