Add --retry argument to control model retry attempts on failure

videogen

@@ -3325,7 +3325,8 @@ def main(args):
         # Store original args for retry
         args._auto_mode = True
         args._retry_count = 0
-        args._max_retries = 3  # Maximum number of model retries
+        # Use --retry argument or default to 3
+        args._max_retries = getattr(args, 'retry', 3)
 
     if args.distribute and args.interface:
         os.environ["NCCL_SOCKET_IFNAME"] = args.interface
@@ -4247,6 +4248,9 @@ List TTS voices:
     # Auto mode arguments
     parser.add_argument("--auto", action="store_true",
                         help="Automatic mode: detect generation type and NSFW from prompts, select best models automatically")
+    parser.add_argument("--retry", type=int, default=3,
+                        metavar="COUNT",
+                        help="Number of retry attempts with alternative models when a model fails (default: 3, use 0 to disable)")
     parser.add_argument("--prefer-speed", action="store_true",
                         help="In auto mode, prefer faster models over higher quality")