Spaces:
Running
Running
Update gradio/vace.py
Browse files- gradio/vace.py +2 -1
gradio/vace.py
CHANGED
|
@@ -393,7 +393,8 @@ if __name__ == '__main__':
|
|
| 393 |
parser.add_argument('--root_path', dest='root_path', help='', default=None)
|
| 394 |
parser.add_argument('--save_dir', dest='save_dir', help='', default='cache')
|
| 395 |
parser.add_argument("--mp", action="store_true", help="Use Multi-GPUs",)
|
| 396 |
-
|
|
|
|
| 397 |
parser.add_argument("--ulysses_size", type=int, default=1, help="The size of the ulysses parallelism in DiT.")
|
| 398 |
parser.add_argument("--ring_size", type=int, default=1, help="The size of the ring attention parallelism in DiT.")
|
| 399 |
parser.add_argument(
|
|
|
|
| 393 |
parser.add_argument('--root_path', dest='root_path', help='', default=None)
|
| 394 |
parser.add_argument('--save_dir', dest='save_dir', help='', default='cache')
|
| 395 |
parser.add_argument("--mp", action="store_true", help="Use Multi-GPUs",)
|
| 396 |
+
if not is_shared_ui and is_gpu_associated:
|
| 397 |
+
parser.add_argument("--model_name", type=str, default="vace-1.3B", choices=list(WAN_CONFIGS.keys()), help="The model name to run.")
|
| 398 |
parser.add_argument("--ulysses_size", type=int, default=1, help="The size of the ulysses parallelism in DiT.")
|
| 399 |
parser.add_argument("--ring_size", type=int, default=1, help="The size of the ring attention parallelism in DiT.")
|
| 400 |
parser.add_argument(
|