Push model using huggingface_hub.

README.md

@@ -0,0 +1,12 @@
+---
+license: mit
+pipeline_tag: image-to-image
+tags:
+- model_hub_mixin
+- pytorch_model_hub_mixin
+---
+
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Code: https://github.com/KhoiDOO/vitvqganvae
+- Paper: [More Information Needed]
+- Docs: https://github.com/KhoiDOO/vitvqganvae

config.json

@@ -0,0 +1,30 @@
+{
+  "codebook_size": 512,
+  "conv_type": "conv1d",
+  "dec_act_func": "GLU",
+  "dec_act_kwargs": {
+    "dim": 1
+  },
+  "dim": 128,
+  "enc_act_func": "LeakyReLU",
+  "enc_act_kwargs": {
+    "negative_slope": 0.1
+  },
+  "first_conv_kernel_size": 5,
+  "group": 8,
+  "in_channel": 3,
+  "l2_recon_loss": true,
+  "layer_mults": null,
+  "layers": 2,
+  "num_res_blocks": 1,
+  "out_channel": 3,
+  "quantizer": "VectorQuantize",
+  "quantizer_kwargs": {
+    "codebook_dim": 64,
+    "commitment_weight": 0.25,
+    "decay": 0.99,
+    "heads": 8,
+    "kmeans_init": true,
+    "use_cosine_sim": true
+  }
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e7606cb86e6814feaf24b9a68b56d0ac5533832e2d21d3b33b158e79c8aecae4
+size 8164048