Update train.py

This commit is contained in:
Victor Hall 2023-01-10 17:28:31 -05:00 committed by GitHub
parent 3519dd6ba0
commit 8a528e9ab1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -842,7 +842,7 @@ def update_old_args(t_args):
t_args.__dict__["shuffle_tags"] = False
if __name__ == "__main__":
supported_resolutions = [384, 448, 512, 576, 640, 704, 768, 832, 896, 960, 1024, 1088, 1152]
supported_resolutions = [256, 384, 448, 512, 576, 640, 704, 768, 832, 896, 960, 1024, 1088, 1152]
argparser = argparse.ArgumentParser(description="EveryDream2 Training options")
argparser.add_argument("--config", type=str, required=False, default=None, help="JSON config file to load options from")
args, _ = argparser.parse_known_args()