Spaces:
Sleeping
Sleeping
Kyryll Kochkin commited on
Commit ·
566c6e8
1
Parent(s): ad9ba57
added trained models
Browse files- __pycache__/dataset.cpython-312.pyc +0 -0
- __pycache__/dataset_vq.cpython-312.pyc +0 -0
- __pycache__/vq_transformer.cpython-312.pyc +0 -0
- __pycache__/vq_vae.cpython-312.pyc +0 -0
- data/MNIST/raw/t10k-images-idx3-ubyte +3 -0
- data/MNIST/raw/t10k-images-idx3-ubyte.gz +3 -0
- data/MNIST/raw/t10k-labels-idx1-ubyte +3 -0
- data/MNIST/raw/t10k-labels-idx1-ubyte.gz +3 -0
- data/MNIST/raw/train-images-idx3-ubyte +3 -0
- data/MNIST/raw/train-images-idx3-ubyte.gz +3 -0
- data/MNIST/raw/train-labels-idx1-ubyte +3 -0
- data/MNIST/raw/train-labels-idx1-ubyte.gz +3 -0
- my_conv/config.json +9 -0
- my_conv/model.safetensors +3 -0
- my_diffusion_model/model_index.json +12 -0
- my_diffusion_model/scheduler/scheduler_config.json +19 -0
- my_diffusion_model/unet/config.json +42 -0
- my_diffusion_model/unet/diffusion_pytorch_model.safetensors +3 -0
- my_model/config.pt +3 -0
- my_model/model.pt +3 -0
- my_moe_model/config.pt +3 -0
- my_moe_model/model.pt +3 -0
- vq_transformer_model/config.pt +3 -0
- vq_transformer_model/model.pt +3 -0
- vq_vae_model.pt +3 -0
- vq_vae_reconstruction.png +0 -0
__pycache__/dataset.cpython-312.pyc
ADDED
|
Binary file (1.84 kB). View file
|
|
|
__pycache__/dataset_vq.cpython-312.pyc
ADDED
|
Binary file (2.93 kB). View file
|
|
|
__pycache__/vq_transformer.cpython-312.pyc
ADDED
|
Binary file (13 kB). View file
|
|
|
__pycache__/vq_vae.cpython-312.pyc
ADDED
|
Binary file (8.76 kB). View file
|
|
|
data/MNIST/raw/t10k-images-idx3-ubyte
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0fa7898d509279e482958e8ce81c8e77db3f2f8254e26661ceb7762c4d494ce7
|
| 3 |
+
size 7840016
|
data/MNIST/raw/t10k-images-idx3-ubyte.gz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8d422c7b0a1c1c79245a5bcf07fe86e33eeafee792b84584aec276f5a2dbc4e6
|
| 3 |
+
size 1648877
|
data/MNIST/raw/t10k-labels-idx1-ubyte
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ff7bcfd416de33731a308c3f266cc351222c34898ecbeaf847f06e48f7ec33f2
|
| 3 |
+
size 10008
|
data/MNIST/raw/t10k-labels-idx1-ubyte.gz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f7ae60f92e00ec6debd23a6088c31dbd2371eca3ffa0defaefb259924204aec6
|
| 3 |
+
size 4542
|
data/MNIST/raw/train-images-idx3-ubyte
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ba891046e6505d7aadcbbe25680a0738ad16aec93bde7f9b65e87a2fc25776db
|
| 3 |
+
size 47040016
|
data/MNIST/raw/train-images-idx3-ubyte.gz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:440fcabf73cc546fa21475e81ea370265605f56be210a4024d2ca8f203523609
|
| 3 |
+
size 9912422
|
data/MNIST/raw/train-labels-idx1-ubyte
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:65a50cbbf4e906d70832878ad85ccda5333a97f0f4c3dd2ef09a8a9eef7101c5
|
| 3 |
+
size 60008
|
data/MNIST/raw/train-labels-idx1-ubyte.gz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3552534a0a558bbed6aed32b30c495cca23d567ec52cac8be1a0730e8010255c
|
| 3 |
+
size 28881
|
my_conv/config.json
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"ConvGeneratorModel"
|
| 4 |
+
],
|
| 5 |
+
"latent_dim": 100,
|
| 6 |
+
"model_type": "conv_generator",
|
| 7 |
+
"torch_dtype": "float32",
|
| 8 |
+
"transformers_version": "4.48.1"
|
| 9 |
+
}
|
my_conv/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:edfd20fcf04b3c7eac334e14f4c9a1670c6a2ab312dcef85ffe17991028ce1e0
|
| 3 |
+
size 4315944
|
my_diffusion_model/model_index.json
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "DDPMPipeline",
|
| 3 |
+
"_diffusers_version": "0.29.0",
|
| 4 |
+
"scheduler": [
|
| 5 |
+
"diffusers",
|
| 6 |
+
"DDPMScheduler"
|
| 7 |
+
],
|
| 8 |
+
"unet": [
|
| 9 |
+
"diffusers",
|
| 10 |
+
"UNet2DModel"
|
| 11 |
+
]
|
| 12 |
+
}
|
my_diffusion_model/scheduler/scheduler_config.json
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "DDPMScheduler",
|
| 3 |
+
"_diffusers_version": "0.29.0",
|
| 4 |
+
"beta_end": 0.02,
|
| 5 |
+
"beta_schedule": "linear",
|
| 6 |
+
"beta_start": 0.0001,
|
| 7 |
+
"clip_sample": true,
|
| 8 |
+
"clip_sample_range": 1.0,
|
| 9 |
+
"dynamic_thresholding_ratio": 0.995,
|
| 10 |
+
"num_train_timesteps": 1000,
|
| 11 |
+
"prediction_type": "epsilon",
|
| 12 |
+
"rescale_betas_zero_snr": false,
|
| 13 |
+
"sample_max_value": 1.0,
|
| 14 |
+
"steps_offset": 0,
|
| 15 |
+
"thresholding": false,
|
| 16 |
+
"timestep_spacing": "leading",
|
| 17 |
+
"trained_betas": null,
|
| 18 |
+
"variance_type": "fixed_small"
|
| 19 |
+
}
|
my_diffusion_model/unet/config.json
ADDED
|
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_class_name": "UNet2DModel",
|
| 3 |
+
"_diffusers_version": "0.29.0",
|
| 4 |
+
"act_fn": "silu",
|
| 5 |
+
"add_attention": true,
|
| 6 |
+
"attention_head_dim": 8,
|
| 7 |
+
"attn_norm_num_groups": null,
|
| 8 |
+
"block_out_channels": [
|
| 9 |
+
32,
|
| 10 |
+
64,
|
| 11 |
+
128
|
| 12 |
+
],
|
| 13 |
+
"center_input_sample": false,
|
| 14 |
+
"class_embed_type": null,
|
| 15 |
+
"down_block_types": [
|
| 16 |
+
"DownBlock2D",
|
| 17 |
+
"AttnDownBlock2D",
|
| 18 |
+
"DownBlock2D"
|
| 19 |
+
],
|
| 20 |
+
"downsample_padding": 1,
|
| 21 |
+
"downsample_type": "conv",
|
| 22 |
+
"dropout": 0.0,
|
| 23 |
+
"flip_sin_to_cos": true,
|
| 24 |
+
"freq_shift": 0,
|
| 25 |
+
"in_channels": 1,
|
| 26 |
+
"layers_per_block": 2,
|
| 27 |
+
"mid_block_scale_factor": 1,
|
| 28 |
+
"norm_eps": 1e-05,
|
| 29 |
+
"norm_num_groups": 32,
|
| 30 |
+
"num_class_embeds": 10,
|
| 31 |
+
"num_train_timesteps": null,
|
| 32 |
+
"out_channels": 1,
|
| 33 |
+
"resnet_time_scale_shift": "default",
|
| 34 |
+
"sample_size": 28,
|
| 35 |
+
"time_embedding_type": "positional",
|
| 36 |
+
"up_block_types": [
|
| 37 |
+
"UpBlock2D",
|
| 38 |
+
"AttnUpBlock2D",
|
| 39 |
+
"UpBlock2D"
|
| 40 |
+
],
|
| 41 |
+
"upsample_type": "conv"
|
| 42 |
+
}
|
my_diffusion_model/unet/diffusion_pytorch_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e16c8bd7078152aaee33ac7dc503a965a35dc3821757e573e8a26b3acbf2fe00
|
| 3 |
+
size 14799964
|
my_model/config.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5250cf8be3e7f548c7e153264adaa938430bf85c366a6be914229ced6d43ebc5
|
| 3 |
+
size 1052
|
my_model/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a8a4d23ff5cbe4ab580cac123949c4a886ba171167fb4db0d5b474b5e0c207b4
|
| 3 |
+
size 45792242
|
my_moe_model/config.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:95613b7796fd7bd40365929eeb8e9b69c9ab93519a7186b32e1bc00b95ec0735
|
| 3 |
+
size 1180
|
my_moe_model/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f81d8804ea122de916808ccd5c02e22259a5f7569476c8318cdb87afce984e3e
|
| 3 |
+
size 180491362
|
vq_transformer_model/config.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:039711d65d809e22b49af2a55d4f0a6786c29d3a90916577002c5fcef4cb91d6
|
| 3 |
+
size 1116
|
vq_transformer_model/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1c8818673ab2c30c139a3f51ca6516beb74c32b93b9e2d43b6ec65b373971f38
|
| 3 |
+
size 1135532
|
vq_vae_model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c4280873f91dcfaa6a6a1f33071c315da539a78aaa14655ed45019df262765de
|
| 3 |
+
size 179524
|
vq_vae_reconstruction.png
ADDED
|