Update train.py

This commit is contained in:
Victor Hall 2023-01-10 17:27:38 -05:00 committed by GitHub
parent 9a9dd8eb3d
commit 3519dd6ba0
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 t_args.__dict__["shuffle_tags"] = False
if __name__ == "__main__": if __name__ == "__main__":
supported_resolutions = [448, 512, 576, 640, 704, 768, 832, 896, 960, 1024, 1088, 1152] supported_resolutions = [384, 448, 512, 576, 640, 704, 768, 832, 896, 960, 1024, 1088, 1152]
argparser = argparse.ArgumentParser(description="EveryDream2 Training options") 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") argparser.add_argument("--config", type=str, required=False, default=None, help="JSON config file to load options from")
args, _ = argparser.parse_known_args() args, _ = argparser.parse_known_args()