Kemajuan ilmu teknologi informasi dan komunikasi (TIK) sudah membawa aspek-aspek kehidupan menuju era baru. Terdapat banyak hal yang terjadi di dunia berlangsung dengan cepat dan secara modern. AI (Artificial Intelligence) Image Generator adalah sebuah tool AI dan di desain untuk menghasilkan gambar yang berkualitas dan realistis berdasarkan deskripsi tekstual atau prompts. Dalam tugas akhir ini, digunakan diffusion dan metode Stochastic Gradient Descent with momentum (SGDM) untuk menghasilkan gambar berupa digit angka. Berdasarkan percobaan yang dilakukan, model loss yang dihasilkan tidak berhasil convergence, dan gambar-gambar yang dihasilkan sama sekali tidak mirip dengan digit angka pada umumnya.