Upload DDPMPipeline

logs/ddpm-paintings-128-finetuned-celebahq/events.out.tfevents.1701751022.coffee.35880.0

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

logs/ddpm-paintings-128-finetuned-celebahq/events.out.tfevents.1701751066.coffee.35898.0

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

main.py

@@ -90,25 +90,26 @@ if __name__ == '__main__':
         'google/ddpm-celebahq-256'
     )
     """
-
     """
-    unet=UNet2DModel.from_pretrained(
-        'jmemon/ddpm-paintings-128-finetuned-celebahq'
+    unet = UNet2DModel.from_pretrained(
+        'jmemon/ddpm-paintings-128-finetuned-celebahq',
+        use_safetensors=True
     ).to('mps')
 
     scheduler = DDPMScheduler.from_pretrained(
         'jmemon/ddpm-paintings-128-finetuned-celebahq'
     )
     """
-
+    
     unet = UNet2DModel.from_pretrained(
-        str(Path(__file__).parent / 'unet')
+        str(Path(__file__).parent / 'unet'),
+        use_safetensors=True
     ).to('mps')
 
     scheduler = DDPMScheduler.from_pretrained(
         str(Path(__file__).parent / 'scheduler')
     )
-
+    
     lora_config = LoraConfig(
         r=8, 
         lora_alpha=8, 
@@ -141,7 +142,7 @@ if __name__ == '__main__':
 
     global_step = 0
     
-    for epoch in range(3, config.num_epochs + 3):
+    for epoch in range(6, config.num_epochs + 6):
         pbar = tqdm(total=len(train_dataloader), disable=not accelerator.is_local_main_process)
         pbar.set_description(f'Epoch {epoch}')
 
@@ -204,7 +205,7 @@ if __name__ == '__main__':
                     model_loc = 'jmemon/ddpm-paintings-128-finetuned-celebahq'
                 else:
                     _pipeline.save_pretrained(config.output_dir)
-                    model_loc = str(Path(__file__).parent / 'diffusion_model_pytorch.bin')
+                    model_loc = str(Path(__file__).parent / 'diffusion_model_pytorch.safetensors')
 
-                unet = UNet2DModel.from_pretrained(model_loc)
+                unet = UNet2DModel.from_pretrained(model_loc, use_safetensors=True)
                 lora_unet = get_peft_model(unet, lora_config)

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b749f8f5afcbd0a78881e2e6fecb6b75b2834f61a29fdea56180a1f71d52fcb3
+oid sha256:3a9bda38d316357ab00e56ccd00fec9186f490a6e9e91dcbc7b504735faf45ea
 size 454741108