Files: Epoch -1

ddpm-paintings-128-finetuned-cifar10/logs/ddpm-paintings-128-finetuned-cifar10/events.out.tfevents.1701709707.coffee.18461.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9dce021c919f28197d4d63e701a42d8b148b776b40d4792e54f17d198c5673a2
+size 88

ddpm-paintings-128-finetuned-cifar10/logs/ddpm-paintings-128-finetuned-cifar10/events.out.tfevents.1701710486.coffee.18593.0

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

main.py

@@ -93,7 +93,7 @@ if __name__ == '__main__':
     lora_config = LoraConfig(
         r=8, 
         lora_alpha=8, 
-        modules_to_save=['unet'], 
+        #modules_to_save=['model'], 
         target_modules=['to_k','to_v'], 
         lora_dropout=0.1, 
         bias='none')
@@ -133,13 +133,15 @@ if __name__ == '__main__':
         token='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'
     )
 
-    pipeline.save_pretrained(
+    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