Anthoniomada commited on
Commit
a726f21
·
verified ·
1 Parent(s): 3ef2193

Update run_diffueraser.py

Browse files
Files changed (1) hide show
  1. run_diffueraser.py +2 -2
run_diffueraser.py CHANGED
@@ -16,8 +16,8 @@ def main():
16
  parser.add_argument('--max_img_size', type=int, default=1050, help='The maximum length of output width and height')
17
  parser.add_argument('--save_path', type=str, default="results" , help='Path to the output')
18
  parser.add_argument('--ref_stride', type=int, default=10, help='Propainter params')
19
- parser.add_argument('--neighbor_length', type=int, default=10, help='Propainter params')
20
- parser.add_argument('--subvideo_length', type=int, default=50, help='Propainter params')
21
  parser.add_argument('--base_model_path', type=str, default="weights/stable-diffusion-v1-5" , help='Path to sd1.5 base model')
22
  parser.add_argument('--vae_path', type=str, default="weights/sd-vae-ft-mse" , help='Path to vae')
23
  parser.add_argument('--diffueraser_path', type=str, default="weights/diffuEraser" , help='Path to DiffuEraser')
 
16
  parser.add_argument('--max_img_size', type=int, default=1050, help='The maximum length of output width and height')
17
  parser.add_argument('--save_path', type=str, default="results" , help='Path to the output')
18
  parser.add_argument('--ref_stride', type=int, default=10, help='Propainter params')
19
+ parser.add_argument('--neighbor_length', type=int, default=20, help='Propainter params')
20
+ parser.add_argument('--subvideo_length', type=int, default=60, help='Propainter params')
21
  parser.add_argument('--base_model_path', type=str, default="weights/stable-diffusion-v1-5" , help='Path to sd1.5 base model')
22
  parser.add_argument('--vae_path', type=str, default="weights/sd-vae-ft-mse" , help='Path to vae')
23
  parser.add_argument('--diffueraser_path', type=str, default="weights/diffuEraser" , help='Path to DiffuEraser')