pharmapsychotic
commited on
Commit
•
cc2be96
1
Parent(s):
0004b4c
pretrained cat models
Browse files- cats512_vae.yaml +11 -0
- cats512_vae_217.safetensors +3 -0
- cats64.yaml +11 -0
- cats64_2738.safetensors +3 -0
cats512_vae.yaml
ADDED
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# pixel resolution of the model
|
2 |
+
resolution: 512
|
3 |
+
|
4 |
+
# train 4 channel VAE latents instead of RGB at 1/8 resolution
|
5 |
+
vae_latents: true
|
6 |
+
|
7 |
+
# number of channels at each level of the UNet
|
8 |
+
channels: [128, 256, 384, 512]
|
9 |
+
|
10 |
+
# which levels have attention
|
11 |
+
attention: [0, 1, 2]
|
cats512_vae_217.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:689634a3df1722eaa61c6a980872311dd0524f766af699b9fb1c189bc3c0766c
|
3 |
+
size 169268456
|
cats64.yaml
ADDED
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# pixel resolution of the model
|
2 |
+
resolution: 64
|
3 |
+
|
4 |
+
# train 4 channel VAE latents instead of RGB at 1/8 resolution
|
5 |
+
vae_latents: false
|
6 |
+
|
7 |
+
# number of channels at each level of the UNet
|
8 |
+
channels: [128, 256, 384, 512]
|
9 |
+
|
10 |
+
# which levels have attention
|
11 |
+
attention: [1, 2]
|
cats64_2738.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1ed46de084961b5dc7a100f8ce5118ef769fb11f063b5bd232e6afd9b82f472f
|
3 |
+
size 168595446
|