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

FLUX: dreambooth add main() def

Browse Source
This commit is contained in:
madroid 2024-11-07 15:20:58 +08:00
parent 39fd6d272f
commit 230215a50d
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
flux/mlx_flux/dreambooth.py
View File

@ -155,7 +155,7 @@ def setup_arg_parser():
return parser
if __name__ == "__main__":
def main():
parser = setup_arg_parser()
args = parser.parse_args()
@ -285,3 +285,7 @@ if __name__ == "__main__":
if (i + 1) % 10 == 0:
losses = []
tic = time.time()
if __name__ == "__main__":
main()