JuyeopDang commited on
Commit
08b94bc
·
verified ·
1 Parent(s): f6fe912

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +52 -3
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.*