Upload folder using huggingface_hub

.gitignore

@@ -0,0 +1,2 @@
+.DS_Store
+logs/

config.py

@@ -13,7 +13,7 @@ class TrainingConfig:
     save_image_epochs = 1
     save_model_epochs = 3
     mixed_precision = 'fp16'  # `no` for float32, `fp16` for automatic mixed precision
-    output_dir = 'ddpm-paintings-128-finetuned-cifar10'  # the model name locally and on the HF Hub
+    output_dir = '.'
 
     push_to_hub = True  # whether to upload the saved model to the HF Hub
     hub_model_id = 'jmemon/ddpm-paintings-128-finetuned-cifar10'  # the name of the repository to create on the HF Hub

logs/ddpm-paintings-128-finetuned-cifar10/events.out.tfevents.1701696166.coffee.14798.0

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

logs/ddpm-paintings-128-finetuned-cifar10/events.out.tfevents.1701704512.coffee.17529.0

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

logs/ddpm-paintings-128-finetuned-cifar10/events.out.tfevents.1701706238.coffee.17871.0

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

logs/ddpm-paintings-128-finetuned-cifar10/events.out.tfevents.1701706320.coffee.17904.0

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

logs/ddpm-paintings-128-finetuned-cifar10/events.out.tfevents.1701706355.coffee.17921.0

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

logs/ddpm-paintings-128-finetuned-cifar10/events.out.tfevents.1701706396.coffee.17940.0

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

logs/ddpm-paintings-128-finetuned-cifar10/events.out.tfevents.1701706427.coffee.17968.0

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

logs/ddpm-paintings-128-finetuned-cifar10/events.out.tfevents.1701706694.coffee.18009.0

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

logs/ddpm-paintings-128-finetuned-cifar10/events.out.tfevents.1701706759.coffee.18061.0

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

logs/ddpm-paintings-128-finetuned-cifar10/events.out.tfevents.1701709667.coffee.18436.0

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

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

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

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:d98b7e9040a9c3d71d7737d66be4b6a155b4d1445c965788a4a58883aff70800
+size 856186

logs/ddpm-paintings-128-finetuned-cifar10/events.out.tfevents.1701722303.coffee.24669.0

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

main.py

@@ -85,11 +85,19 @@ if __name__ == '__main__':
         'google/ddpm-celebahq-256',
         safetensors=True
     ).to('mps')
-
+    
     scheduler = DDPMScheduler.from_pretrained(
         'google/ddpm-celebahq-256'
     )
 
+    """unet=UNet2DModel.from_pretrained(
+        'jmemon/ddpm-paintings-128-finetuned-celebahq'
+    ).to('mps')
+
+    scheduler = DDPMScheduler.from_pretrained(
+        'jmemon/ddpm-paintings-128-finetuned-celebahq'
+    )"""
+
     lora_config = LoraConfig(
         r=8, 
         lora_alpha=8, 

model/adapter_model.bin

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

model/model_index.json

@@ -0,0 +1,12 @@
+{
+  "_class_name": "DDPMPipeline",
+  "_diffusers_version": "0.24.0",
+  "scheduler": [
+    "diffusers",
+    "DDPMScheduler"
+  ],
+  "unet": [
+    null,
+    null
+  ]
+}

model/scheduler/scheduler_config.json

@@ -0,0 +1,18 @@
+{
+  "_class_name": "DDPMScheduler",
+  "_diffusers_version": "0.24.0",
+  "beta_end": 0.02,
+  "beta_schedule": "linear",
+  "beta_start": 0.0001,
+  "clip_sample": true,
+  "clip_sample_range": 1.0,
+  "dynamic_thresholding_ratio": 0.995,
+  "num_train_timesteps": 1000,
+  "prediction_type": "epsilon",
+  "sample_max_value": 1.0,
+  "steps_offset": 0,
+  "thresholding": false,
+  "timestep_spacing": "leading",
+  "trained_betas": null,
+  "variance_type": "fixed_small"
+}

test.py

@@ -0,0 +1,23 @@
+from pathlib import Path
+
+from diffusers import DDPMPipeline, UNet2DModel, DDPMScheduler
+from diffusers.utils import make_image_grid
+import torch
+
+
+if __name__ == '__main__':
+
+    unet=UNet2DModel.from_pretrained(
+        'jmemon/ddpm-paintings-128-finetuned-celebahq'
+    )
+
+    scheduler = DDPMScheduler.from_pretrained(
+        'jmemon/ddpm-paintings-128-finetuned-celebahq'
+    )
+
+    pipeline = DDPMPipeline(unet=unet, scheduler=scheduler).to('mps')
+    pipeline.enable_attention_slicing()
+
+    images = pipeline(batch_size=4, generator=torch.manual_seed(0), num_inference_steps=50).images
+    grid = make_image_grid(images, 2, 2)
+    grid.save(Path(__file__).parent / 'grid.png')