Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,52 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: mit
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
tags:
|
| 4 |
+
- diffusion
|
| 5 |
+
- latent-diffusion
|
| 6 |
+
- pytorch
|
| 7 |
+
- text-to-image
|
| 8 |
+
- image-generation
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# Simple Latent Diffusion Model (LDM)
|
| 12 |
+
|
| 13 |
+
This repository contains the pre-trained weights and configuration files for the **Simple Latent Diffusion Model** project.
|
| 14 |
+
|
| 15 |
+
For the full source code, detailed explanations, and implementation logic, please visit the [original GitHub repository](https://github.com/Won-Seong/simple-latent-diffusion-model).
|
| 16 |
+
|
| 17 |
+
## 🚀 Model Description
|
| 18 |
+
This project implements a **Latent Diffusion Model (LDM)** from scratch. The repository includes:
|
| 19 |
+
* **Custom-trained VAE**: For compressing images into a latent space.
|
| 20 |
+
* **Diffusion Model**: A U-Net based architecture for the reverse diffusion process.
|
| 21 |
+
* **CLIP Weights**: Integrated for text-guided image generation.
|
| 22 |
+
|
| 23 |
+
---
|
| 24 |
+
|
| 25 |
+
## 📂 Available Models & Checkpoints
|
| 26 |
+
|
| 27 |
+
The repository provides weights for three different datasets, covering both unconditional and conditional generation tasks:
|
| 28 |
+
|
| 29 |
+
| Dataset | Type | Description |
|
| 30 |
+
| :--- | :--- | :--- |
|
| 31 |
+
| **CIFAR-10** | Unconditional | 32x32 image generation based on CIFAR-10 classes. |
|
| 32 |
+
| **CelebA** | Unconditional | Human face generation trained on the CelebA dataset. |
|
| 33 |
+
| **Asian Composite** | Text-to-Image (T2I) | CLIP-based conditional generation using the Asian Composite Dataset. |
|
| 34 |
+
|
| 35 |
+
---
|
| 36 |
+
|
| 37 |
+
## 🛠 How to Use
|
| 38 |
+
If you want to experiment with these models and generate your own images, we provide a hands-on example notebook.
|
| 39 |
+
|
| 40 |
+
1. Open the **`cifar10_example.ipynb`** file provided in this repository.
|
| 41 |
+
2. Follow the step-by-step instructions to load the configurations and model weights.
|
| 42 |
+
3. Run the cells to start the sampling process and generate images.
|
| 43 |
+
|
| 44 |
+
---
|
| 45 |
+
|
| 46 |
+
## 🔗 References
|
| 47 |
+
* **GitHub Repository**: [Won-Seong/simple-latent-diffusion-model](https://github.com/Won-Seong/simple-latent-diffusion-model)
|
| 48 |
+
* **Contact**: For detailed code logic or issues, please refer to the GitHub documentation or open an issue there.
|
| 49 |
+
|
| 50 |
+
---
|
| 51 |
+
|
| 52 |
+
**Note**: *Ensure you have the necessary dependencies installed (PyTorch, Diffusers, Transformers, etc.) as specified in the GitHub repository's requirements.*
|