4eJIoBek commited on
Commit
829c94d
1 Parent(s): 31aa0a4

Update demo_web.py

Browse files
Files changed (1) hide show
  1. 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="bes-dev/stable-diffusion-v1-4-openvino", 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")
 
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")