zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-30 19:06:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

FLUX: split small functions

Browse Source
This commit is contained in:
madroid 2024-10-12 18:44:47 +08:00
parent a5f2bab070
commit b0e017a16c
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
flux/dreambooth.py
View File

@ -157,7 +157,8 @@ def save_adapters(iteration, flux, args):
) )
if __name__ == "__main__": def setup_arg_parser():
"""Set up and return the argument parser."""
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description="Finetune Flux to generate images with a specific subject" description="Finetune Flux to generate images with a specific subject"
) )
@ -247,7 +248,11 @@ if __name__ == "__main__":
) )
parser.add_argument("dataset") parser.add_argument("dataset")
return parser
if __name__ == "__main__":
parser = setup_arg_parser()
args = parser.parse_args() args = parser.parse_args()
# Load the model and set it up for LoRA training. We use the same random # Load the model and set it up for LoRA training. We use the same random