From 515f0f1fdc9a76bf63bd746c291dcfec7fc747fb Mon Sep 17 00:00:00 2001 From: Volpeon Date: Thu, 13 Oct 2022 21:11:53 +0200 Subject: Added support for Aesthetic Gradients --- textual_inversion.py | 10 ---------- 1 file changed, 10 deletions(-) (limited to 'textual_inversion.py') diff --git a/textual_inversion.py b/textual_inversion.py index 181a318..9d2840d 100644 --- a/textual_inversion.py +++ b/textual_inversion.py @@ -192,12 +192,6 @@ def parse_args(): "and an Nvidia Ampere GPU." ), ) - parser.add_argument( - "--local_rank", - type=int, - default=-1, - help="For distributed training: local_rank" - ) parser.add_argument( "--checkpoint_frequency", type=int, @@ -280,10 +274,6 @@ def parse_args(): args = parser.parse_args( namespace=argparse.Namespace(**json.load(f)["args"])) - env_local_rank = int(os.environ.get("LOCAL_RANK", -1)) - if env_local_rank != -1 and env_local_rank != args.local_rank: - args.local_rank = env_local_rank - if args.train_data_file is None: raise ValueError("You must specify --train_data_file") -- cgit v1.2.3-54-g00ecf