https://github.com/neelr/scratche-dit

DiT impementation!
https://github.com/neelr/scratche-dit

Last synced: 2 months ago
JSON representation

DiT impementation!

• Host: GitHub
• URL: https://github.com/neelr/scratche-dit
• Owner: neelr
• Created: 2024-08-26T06:38:32.000Z (9 months ago)
• Default Branch: main
• Last Pushed: 2024-09-15T15:05:23.000Z (8 months ago)
• Last Synced: 2025-01-16T20:55:45.637Z (4 months ago)
• Language: Jupyter Notebook
• Size: 1.23 MB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# DiT reproduction from "scratch"!

the goal was to reproduce https://arxiv.org/abs/2212.09748 in the most simple way possible, without using chatgpt/copilot/any online code—only the pytorch docs and myself

the output is a (readable) DiT implementation in ~130 lines of code!

https://neelr.notion.site/pt-1-diffusion-model-from-scratch-202c1ca586fd4cf195ea23ff70975efb?pvs=4