刘虹雨 commited on
Commit
626d0b4
·
1 Parent(s): 2b28eee

update code

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -153,7 +153,7 @@ def get_args():
153
  parser.add_argument("--sampling_algo", type=str, default="dpm-solver",
154
  choices=["iddpm", "dpm-solver"],
155
  help="Sampling algorithm to be used.")
156
- parser.add_argument("--seed", type=int, default=0,
157
  help="Random seed for reproducibility.")
158
  # parser.add_argument("--select_img", type=str, default=None,
159
  # help="Optional: Select a specific image.")
@@ -358,8 +358,6 @@ def launch_pretrained():
358
  )
359
  logging.info("delete models.")
360
 
361
- os.remove('./pretrained_model/sd21/v2-1_512-ema-pruned.ckpt')
362
- os.remove('./pretrained_model/sd21/v2-1_512-nonema-pruned.ckpt')
363
 
364
  # 下载 CrucibleAI/ControlNetMediaPipeFace 的所有文件
365
  snapshot_download(
@@ -852,7 +850,6 @@ def launch_gradio_app():
852
  is_styled = gr.State(value=False)
853
  working_dir = gr.State()
854
  image_name_true = gr.State()
855
- error_box = gr.Textbox(label="error hint", lines=3, interactive=False, visible=True)
856
 
857
 
858
  with gr.Row():
 
153
  parser.add_argument("--sampling_algo", type=str, default="dpm-solver",
154
  choices=["iddpm", "dpm-solver"],
155
  help="Sampling algorithm to be used.")
156
+ parser.add_argument("--seed", type=int, default=42,
157
  help="Random seed for reproducibility.")
158
  # parser.add_argument("--select_img", type=str, default=None,
159
  # help="Optional: Select a specific image.")
 
358
  )
359
  logging.info("delete models.")
360
 
 
 
361
 
362
  # 下载 CrucibleAI/ControlNetMediaPipeFace 的所有文件
363
  snapshot_download(
 
850
  is_styled = gr.State(value=False)
851
  working_dir = gr.State()
852
  image_name_true = gr.State()
 
853
 
854
 
855
  with gr.Row():