Skip to content

Latest commit

 

History

12 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pocket Diffusion πŸš€πŸ”¬πŸ§ 

🀌 Trying to implement some of the Stable Diffusion Concepts and Architectures

Pocket Diffusion is a PyTorch-based implementation of stable diffusion-based models. This repository aims to provide a basic implementation of diffusion models using PyTorch, allowing researchers and developers to experiment with and explore the capabilities of diffusion-based modeling.

Diffusion models have gained significant attention in the field of deep learning due to their ability to generate high-quality samples and perform tasks such as image synthesis, inpainting, and denoising. Pocket Diffusion offers a simple and accessible implementation to get started with diffusion models and understand their underlying principles.

Todo ⬜️

  • Design the Blueprint
    • AutoEncoder - VAE
    • Text Encoder
    • UNet Based Model
  • Getting the Basic parts setup
  • Run a Basic Model
  • Find out Dataset to Run on a small model

In Progress 🚧

  • [ ]
  • [ ]

Done βœ…

  • Creating the Repo

Functionalities πŸ§©πŸ”§πŸŽ¨

  • Prompt to Image Generation πŸ–‹οΈπŸ–ΌοΈ
  • Image to Image Generation Based on Prompts πŸ”„πŸ–ΌοΈπŸ”„
  • Textual Inversion πŸ”€πŸ“
  • Classifier Free Guidance πŸŽ―πŸ†“
  • Image Inpainting πŸ–ŒοΈπŸ–ΌοΈ
  • Image Outpainting πŸ–ŒοΈπŸ”²πŸ–ΌοΈ
  • Control Net eventually πŸ•ΉοΈπŸ“ˆ

Acknowledgments

The implementation in Pocket Diffusion is inspired by the works of diffusion models and various PyTorch-based deep learning projects. We acknowledge the valuable contributions of the open-source community in advancing the field of deep learning.

Original Paper

Feel free to reach out to us with any questions or feedback. Happy exploring with Pocket Diffusion! πŸš€

Disclaimer

There are a lot... commits, please consider them as the continuation of the previous commit with tiny changes. πŸ˜„πŸ˜„πŸ˜„

About

Trying to implement some of the Stable Diffusion Concepts and Architectures

Resources

Stars

3 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors