Skip to content
Snippets Groups Projects
Select Git revision
  • fb2e6b8611d23fbf60be10761a2ff11a3afe8159
  • main default protected
  • in-out-painting
3 results

conditional-diffusion

  • Open with
  • Download source code
  • Your workspaces

      A workspace is a virtual sandbox environment for your code in GitLab.

      No agents available to create workspaces. Please consult Workspaces documentation for troubleshooting.

  • gonzalomartingarcia0's avatar
    Gonzalo Martin Garcia authored
    Completed Class Conditional Diffusion Model. This version is intended to be trained on a class labeled dataset. It makes use of classifier-free guided diffusion to boost sample quality of the model when generating images from each class. The conditioning mechanism in the UNet (implemented together with Roy) simply adds an embedding of the class to the time embedding before passing them through the learnable reshaping layers for each block. Successfully trained on 3 class dataset, dogs, cats and wildlife.
    fb2e6b86
    History