waleko commited on
Commit
ddec885
·
verified ·
1 Parent(s): 18af520

Add model card

Browse files
Files changed (1) hide show
  1. README.md +35 -0
README.md ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ # latent-diffusion-autoencoder-128
3
+
4
+ This is a PyTorch checkpoint for a Latent Diffusion model.
5
+
6
+ ## Model Details
7
+ - **Checkpoint file**: `100000.pth`
8
+ - **File size**: 3086.4 MB
9
+ - **Framework**: PyTorch
10
+ - **Model type**: Latent Diffusion Autoencoder
11
+
12
+ ## Usage
13
+
14
+ ```python
15
+ import torch
16
+ from huggingface_hub import hf_hub_download
17
+
18
+ # Download the checkpoint
19
+ checkpoint_path = hf_hub_download(
20
+ repo_id="YOUR_USERNAME/latent-diffusion-autoencoder-128",
21
+ filename="100000.pth"
22
+ )
23
+
24
+ # Load the checkpoint
25
+ checkpoint = torch.load(checkpoint_path, map_location='cpu')
26
+ print("Checkpoint keys:", list(checkpoint.keys()))
27
+ ```
28
+
29
+ ## Training Details
30
+ - **Dataset**: OpenWebText-512
31
+ - **Latent dimensions**: 128
32
+ - **Training steps**: 100,000
33
+
34
+ ## Citation
35
+ If you use this model, please cite the original Latent Diffusion paper and this checkpoint.