End of training

README.md

@@ -14,7 +14,7 @@ inference: true
     
 # Text-to-image finetuning - kerianheYi/CS245-fine-tunedSD100
 
-This pipeline was finetuned from **stabilityai/stable-diffusion-2-base** on the **jytjyt05/t_to_m2** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['None']: 
+This pipeline was finetuned from **stabilityai/stable-diffusion-2-base** on the **jytjyt05/t_to_m2** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['A classical piano solo in C-major']: 
 
 ![val_imgs_grid](./val_imgs_grid.png)
 
@@ -28,7 +28,7 @@ from diffusers import DiffusionPipeline
 import torch
 
 pipeline = DiffusionPipeline.from_pretrained("kerianheYi/CS245-fine-tunedSD100", torch_dtype=torch.float16)
-prompt = "None"
+prompt = "A classical piano solo in C-major"
 image = pipeline(prompt).images[0]
 image.save("my_image.png")
 ```
@@ -37,7 +37,7 @@ image.save("my_image.png")
 
 These are the key hyperparameters used during training:
 
-* Epochs: 1
+* Epochs: 4
 * Learning rate: 1e-05
 * Batch size: 1
 * Gradient accumulation steps: 4

logs/text2image-fine-tune/1701127851.6032078/events.out.tfevents.1701127851.251c89de5e85.21171.1

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

logs/text2image-fine-tune/1701127851.6048746/hparams.yml

@@ -0,0 +1,51 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.01
+allow_tf32: false
+cache_dir: null
+caption_column: caption
+center_crop: false
+checkpointing_steps: 100000
+checkpoints_total_limit: null
+dataloader_num_workers: 0
+dataset_config_name: null
+dataset_name: jytjyt05/t_to_m2
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: true
+hub_model_id: null
+hub_token: null
+image_column: image
+input_perturbation: 0
+learning_rate: 1.0e-05
+local_rank: -1
+logging_dir: logs
+lr_scheduler: constant
+lr_warmup_steps: 0
+max_grad_norm: 1.0
+max_train_samples: null
+max_train_steps: 1
+mixed_precision: fp16
+noise_offset: 0
+non_ema_revision: null
+num_train_epochs: 1
+output_dir: kerianheyi/CS245-fine-tunedSD100
+prediction_type: null
+pretrained_model_name_or_path: stabilityai/stable-diffusion-2-base
+push_to_hub: true
+random_flip: false
+report_to: tensorboard
+resolution: 512
+resume_from_checkpoint: null
+revision: null
+scale_lr: false
+seed: null
+snr_gamma: null
+tracker_project_name: text2image-fine-tune
+train_batch_size: 1
+train_data_dir: null
+use_8bit_adam: false
+use_ema: true
+validation_epochs: 5
+variant: null

logs/text2image-fine-tune/1701127965.9765682/events.out.tfevents.1701127965.251c89de5e85.21691.1

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

logs/text2image-fine-tune/1701127965.9782622/hparams.yml

@@ -0,0 +1,51 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.01
+allow_tf32: false
+cache_dir: null
+caption_column: caption
+center_crop: false
+checkpointing_steps: 100000
+checkpoints_total_limit: null
+dataloader_num_workers: 0
+dataset_config_name: null
+dataset_name: jytjyt05/t_to_m2
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: true
+hub_model_id: null
+hub_token: null
+image_column: image
+input_perturbation: 0
+learning_rate: 1.0e-05
+local_rank: -1
+logging_dir: logs
+lr_scheduler: constant
+lr_warmup_steps: 0
+max_grad_norm: 1.0
+max_train_samples: null
+max_train_steps: 100
+mixed_precision: fp16
+noise_offset: 0
+non_ema_revision: null
+num_train_epochs: 4
+output_dir: kerianheyi/CS245-fine-tunedSD100
+prediction_type: null
+pretrained_model_name_or_path: stabilityai/stable-diffusion-2-base
+push_to_hub: true
+random_flip: false
+report_to: tensorboard
+resolution: 512
+resume_from_checkpoint: null
+revision: null
+scale_lr: false
+seed: null
+snr_gamma: null
+tracker_project_name: text2image-fine-tune
+train_batch_size: 1
+train_data_dir: null
+use_8bit_adam: false
+use_ema: true
+validation_epochs: 5
+variant: null

logs/text2image-fine-tune/events.out.tfevents.1701127372.251c89de5e85.19106.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8a9fd30c67c9d888bf25322ae8d510d42880a4cefde3c6dab55154a0eb921f44
-size 344200
+oid sha256:93f71777f8a6a3b4f747cae02a910650169b8665f4d5e381d29e36bf6c38adbb
+size 345363

logs/text2image-fine-tune/events.out.tfevents.1701127851.251c89de5e85.21171.0

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

logs/text2image-fine-tune/events.out.tfevents.1701127965.251c89de5e85.21691.0

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

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1aafcf45d2f709f3dd78e456c3b7080bd35ae156a532cd46678867858e3d7651
+oid sha256:6aabe1549dc388a9b41dc2af9a695425bc30d92e6b1e0c9c19ce807c601672e5
 size 3463726504