Hugging Face's logo

xkronosx
/
train_prior_ldm-ffhq-256

Diffusers
Safetensors
LDMGFNPipeline
Model card Files Community
xkronosx commited on
Commit
5d8e9f1
·
verified ·
1 Parent(s): 28f0aec

Iteration 0

Browse files
Files changed (6) hide show
  1. autoencoder/config.json +28 -7
  2. autoencoder/diffusion_pytorch_model.safetensors +2 -2
  3. checkpoint.tar +2 -2
  4. model_index.json +2 -2
  5. unet/config.json +0 -1
  6. unet/diffusion_pytorch_model.safetensors +1 -1
autoencoder/config.json CHANGED
@@ -1,13 +1,34 @@
1
  {
2
- "_class_name": "ResNetEncoderDecoder",
3
  "_diffusers_version": "0.27.2",
4
- "_name_or_path": "xkronosx/AutoEncoder-ffhq-256",
5
- "base_filters": 32,
 
 
 
 
 
 
 
 
 
 
 
6
  "in_channels": 3,
7
  "latent_channels": 3,
8
- "latent_size": [
9
- 64,
10
- 64
 
 
 
 
 
 
 
 
 
 
11
  ],
12
- "num_blocks": 3
13
  }
 
1
  {
2
+ "_class_name": "VQModel",
3
  "_diffusers_version": "0.27.2",
4
+ "_name_or_path": "/home/mila/l/luca.scimeca/.cache/huggingface/hub/models--CompVis--ldm-celebahq-256/snapshots/03978f22272a3c2502da709c3940e227c9714bdd/vqvae",
5
+ "act_fn": "silu",
6
+ "block_out_channels": [
7
+ 128,
8
+ 256,
9
+ 512
10
+ ],
11
+ "down_block_types": [
12
+ "DownEncoderBlock2D",
13
+ "DownEncoderBlock2D",
14
+ "DownEncoderBlock2D"
15
+ ],
16
+ "force_upcast": false,
17
  "in_channels": 3,
18
  "latent_channels": 3,
19
+ "layers_per_block": 2,
20
+ "lookup_from_codebook": false,
21
+ "mid_block_add_attention": true,
22
+ "norm_num_groups": 32,
23
+ "norm_type": "group",
24
+ "num_vq_embeddings": 8192,
25
+ "out_channels": 3,
26
+ "sample_size": 256,
27
+ "scaling_factor": 0.18215,
28
+ "up_block_types": [
29
+ "UpDecoderBlock2D",
30
+ "UpDecoderBlock2D",
31
+ "UpDecoderBlock2D"
32
  ],
33
+ "vq_embed_dim": null
34
  }
autoencoder/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2427de83b4b11449e5211efbb88b7f19462ea5fa875533630d75a8bf8d63a34a
3
- size 2129596
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:55cc7d2414ab4dc44da353534cca84f19e08dc4a58b0ca107e4ac7e1eebfbb86
3
+ size 221314016
checkpoint.tar CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b7a10970eac01708e3fa9702c1aeac3f6aa16c1bf51c0ed25ade3fe304e44798
3
- size 127485686
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b07baa640371bcb62b84becf17d9a627fb39de274112c5ff3c02c7b516f269b
3
+ size 127485622
model_index.json CHANGED
@@ -2,8 +2,8 @@
2
  "_class_name": "LDMGFNPipeline",
3
  "_diffusers_version": "0.27.2",
4
  "autoencoder": [
5
- "models.EncoderDecoders",
6
- "ResNetEncoderDecoder"
7
  ],
8
  "scheduler": [
9
  "diffusers",
 
2
  "_class_name": "LDMGFNPipeline",
3
  "_diffusers_version": "0.27.2",
4
  "autoencoder": [
5
+ "diffusers",
6
+ "VQModel"
7
  ],
8
  "scheduler": [
9
  "diffusers",
unet/config.json CHANGED
@@ -1,7 +1,6 @@
1
  {
2
  "_class_name": "UNet2DModel",
3
  "_diffusers_version": "0.27.2",
4
- "_name_or_path": "/network/scratch/l/luca.scimeca/gfn_generated_results/train_prior_ldm-ffhq-256_0/checkpoints/unet",
5
  "act_fn": "silu",
6
  "add_attention": true,
7
  "attention_head_dim": 8,
 
1
  {
2
  "_class_name": "UNet2DModel",
3
  "_diffusers_version": "0.27.2",
 
4
  "act_fn": "silu",
5
  "add_attention": true,
6
  "attention_head_dim": 8,
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:41e4b078798fff2a96abc5007d42366af9058e884fd7ccf09696dca21d045692
3
  size 63612900
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9572da4724ed6ddf4cf3a2931f4cc2fe9b18c2ccc279fa15b6e211f7544d954b
3
  size 63612900