Add correct and remove wrong config file

cog.yaml

@@ -1,21 +0,0 @@
-build:
-  python_version: "3.8"
-
-  gpu: true
-
-  python_packages:
-    - 'huggingface_hub==0.8.1'
-    - 'torch==1.7.1'
-    - 'torchvision==0.8.2'
-    - 'tokenizers==0.10.2'
-    - 'pyflakes==2.2.0'
-    - 'tqdm==4.46.0'
-    - 'pytorch-lightning==1.5'
-    - 'einops==0.4.1'
-    - 'omegaconf==2.2.2'
-    - 'git+https://github.com/openai/CLIP.git'
-    - 'numpy==1.23.1'
-    - 'pillow==9.2.0'
-    - 'python-dotenv==0.20.0'
-
-predict: "predict.py:Predictor"

config.yaml

@@ -0,0 +1,38 @@
+dataset:
+  tokenizer_type: CharBPE
+  context_length: 64
+  image_resolution: 256
+
+stage1:
+  type: vqgan
+  embed_dim: 256
+  n_embed: 16384
+  hparams:
+    double_z: False
+    z_channels: 256
+    resolution: 256
+    in_channels: 3
+    out_ch: 3
+    ch: 128
+    ch_mult: [1, 1, 2, 2, 4]
+    num_res_blocks: 2
+    attn_resolutions: [16]
+    pdrop: 0.0
+
+stage2:
+  type: transformer1d
+  vocab_size_txt: 16384
+  vocab_size_img: 16384
+  hparams:
+    embed_dim: 1536
+    n_layers: 42
+    n_heads: 24
+    n_dense_layers: 42
+    ctx_len_img: 256
+    ctx_len_txt: 64
+    embd_pdrop: 0.0
+    resid_pdrop: 0.0
+    attn_pdrop: 0.0
+    mlp_bias: True
+    attn_bias: True
+    gelu_use_approx: False