fffiloni commited on
Commit
8c83f7a
·
verified ·
1 Parent(s): 84cf951

Update gradio/vace.py

Browse files
Files changed (1) hide show
  1. 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
- parser.add_argument("--model_name", type=str, default="vace-1.3B", choices=list(WAN_CONFIGS.keys()), help="The model name to run.")
 
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(