Commit History
-
Add model card
Verified signature2853c71
-
Add wandb logging (Life is more peacefull)
7e6af10
-
Model not moving forward
bf96a55 -
Add update logic
2e5aaca -
Add scheduler
04897d8 -
Sloving conflicts
88f49f7 -
More stuff
fca91c0 -
Update disc_loss update logic
eec35f5 -
Made changes in the training script
2c4ef68 -
More samples
90d71da -
Update
f4c7526 -
Re training will finish this time for sure
8082844
-
More samples
680c645 -
Halfway through
de7e31f -
Remove all samples
e183f2c
-
Checkpoint for 2nd epoch
1c6f5f9 -
Checkpointnt for 1st epoch
af8cdd6 -
Change checkpoint dir
1a70dc1 -
Modify training code
bba6806 -
Remove logging statements
2473850 -
Move to gpu and add checkpoint images to attributes
f5927bd -
Print statements for logging
43b6c80 -
Add num_heads variables
Yash Nagraj92078be -
Make changes
Yash Nagrajf0ff580 -
Resolve run time errors
8d85e1f -
Make changes to get image
Yash Nagraj789dd08 -
Get the buffers in order to open as image
Yash Nagraj199f027 -
Change any output errors
788a6e1 -
Add im_path from the config
Yash Nagraj1013ca2 -
Change the config file accordingly
Yash Nagraj432f436
-
Add loss printing and saving logic
Yash Nagraj936be8c -
Optimize Discriminator in the training loop
Yash Nagraje57e318 -
Change dataloader loading from directory
Yash Nagrajd67fec6
-
Add dataloader for training
Yash Nagraj4c5a8ab -
Add the dataset
Yash Nagraj2b3a313 -
Move the pretrained vgg16 model to another directory
Yash Nagrajf6182a3 -
Add forward function to the Diffusion model UNet
Yash Nagrajdff68d4 -
Build the Unet from the config
Yash Nagraj6966d99 -
change variable names
Yash Nagraj2e8c272 -
Get all parameters from model_config
Yash Nagraj97679c9 -
Add discriminator for adverserial loss for VAE training
Yash Nagrajec937a4
-
Add lpips for VAE training
Yash Nagrajb311ae5 -
Clean up
Yash Nagraj2a1c7f2 -
Add AutoEncoder (VQVAE)
Yash Nagraj7377e9c
-
Add UpBlock Unet to concat downblock's output
Yash Nagraj70a401a -
Add all the layers in the UpBlock
Yash Nagraj6801d5a
-
Add forward function to the midblock
Yash Nagrajde5e356 -
Add residual convolution to the mid block
Yash Nagraj38d054a
-
Add MidBlocks and change cross_attn in down blocks
Yash Nagrajc2857b5 -
Add cross attn if needed for conditional latent diffusion
Yash Nagraj95b2cf2