Update run_diffueraser.py
Browse files- 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=
|
| 20 |
-
parser.add_argument('--subvideo_length', type=int, default=
|
| 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')
|