End of training

README.md

@@ -29,4 +29,4 @@ These are the key hyperparameters used during training:
 * Mixed-precision: fp16
 
 
-More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/wearesameasyou/vae-fine-tune/runs/55cxj27d).
+More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/wearesameasyou/vae-fine-tune/runs/jil7bkcu).

config.json

@@ -9,14 +9,14 @@
     "DownEncoderBlock2D"
   ],
   "force_upcast": true,
-  "in_channels": 3,
+  "in_channels": 2,
   "latent_channels": 4,
   "latents_mean": null,
   "latents_std": null,
   "layers_per_block": 1,
   "mid_block_add_attention": true,
   "norm_num_groups": 32,
-  "out_channels": 3,
+  "out_channels": 2,
   "sample_size": 32,
   "scaling_factor": 0.18215,
   "shift_factor": null,

diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2c3507df329cea9b2c2360c20e48ba526432221409ab237d2f6d0612cf27f557
-size 2262204
+oid sha256:0658b49da17b33ae31d6ca67120b844001344794923aaeda63d29d87365edf96
+size 2257592