Epoch 2

ddpm-paintings-128-finetuned-cifar10/logs/ddpm-paintings-128-finetuned-cifar10/events.out.tfevents.1701710600.coffee.18648.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:36dbbc8167bb039257d900769381588ef2f7e5ddb372415418c650682f70f494
+size 427942

ddpm-paintings-128-finetuned-cifar10/model/adapter_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16796e38d4a199c3fba38faa6dbeca4ab3186dfcb43bb0a19aa397f28d298c89
+size 455281714

main.py

@@ -121,29 +121,6 @@ if __name__ == '__main__':
 
         accelerator.init_trackers('ddpm-paintings-128-finetuned-cifar10')
 
-
-    epoch = -1
-    pipeline = DDPMPipeline(unet=accelerator.unwrap_model(lora_unet), scheduler=scheduler)
-    
-    upload_folder(
-        repo_id=repo_id,
-        folder_path=Path(config.output_dir).parent,
-        commit_message=f'Files: Epoch {epoch}',
-        ignore_patterns=['logs'],
-        token='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'
-    )
-
-    torch.save(pipeline.unet.state_dict(), Path(config.output_dir) / 'model' / 'adapter_model.bin')
-
-    """pipeline.save_pretrained(
-        str(Path(config.output_dir) / 'model'),
-        #push_to_hub=True,
-        repo_id=config.hub_model_id,
-        commit_message=f'Model: Epoch {epoch}',
-        token='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'
-    )"""
-
-    exit()
     global_step = 0
     
     for epoch in range(config.num_epochs):
@@ -187,18 +164,15 @@ if __name__ == '__main__':
 
             if (epoch + 1) % config.save_model_epochs == 0 or epoch == config.num_epochs - 1:
                 if config.push_to_hub:
+                    torch.save(pipeline.unet.state_dict(), Path(config.output_dir) / 'model' / 'adapter_model.bin')
+
                     upload_folder(
                         repo_id=repo_id,
                         folder_path=Path(config.output_dir).parent,
-                        commit_message=f'Files: Epoch {epoch}',
-                        ignore_patterns=['step_*', 'epoch_*'],
+                        commit_message=f'Epoch {epoch}',
+                        ignore_patterns=['logs'],
                         token='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'
                     )
 
-                    pipeline.push_to_hub(
-                        repo_id=config.hub_model_id,
-                        commit_message=f'Model: Epoch {epoch}',
-                        token='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'
-                    )
                 else:
                     pipeline.save_pretrained(config.output_dir)