junyeong-nero commited on
Commit
ee9ca69
·
verified ·
1 Parent(s): 37aaf0c

Upload vae checkpoint

Browse files
Files changed (2) hide show
  1. README.md +41 -0
  2. vae.pt +3 -0
README.md ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - tiny-stable-diffusion
5
+ - vae
6
+ - image-generation
7
+ - diffusion
8
+ library_name: pytorch
9
+ ---
10
+
11
+ # tiny-sd-models
12
+
13
+ This is a **VAE** model trained with [tiny-stable-diffusion](https://github.com/your-username/tiny-stable-diffusion).
14
+
15
+ ## Model Description
16
+
17
+ This is a Variational Autoencoder (VAE) trained to compress images into a latent space.
18
+ The VAE follows the SD3 architecture with 16 latent channels and f8 compression ratio.
19
+
20
+ ### Architecture
21
+ - **Type**: AutoencoderKL
22
+ - **Latent Channels**: 16
23
+ - **Compression**: f8 (64x64 → 8x8)
24
+
25
+ ## Usage
26
+
27
+ ```python
28
+ import torch
29
+ from src.models.vae import create_vae # or appropriate model import
30
+
31
+ # Load checkpoint
32
+ checkpoint = torch.load("model.pt", map_location="cpu")
33
+
34
+ # Create model and load weights
35
+ model = create_model(...) # Use config from checkpoint
36
+ model.load_state_dict(checkpoint["model_state_dict"])
37
+ ```
38
+
39
+ ## License
40
+
41
+ MIT License
vae.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e98929e0bc3a5209600626aa83707660559d21c6a9c4a5140e34eadc7fbb6473
3
+ size 252594535