Upload RetNetForCausalLM

config.json

@@ -0,0 +1,37 @@
+{
+  "activation_dropout": 0.0,
+  "activation_fn": "swish",
+  "architectures": [
+    "RetNetForCausalLM"
+  ],
+  "decoder_embed_dim": 512,
+  "decoder_ffn_embed_dim": 864,
+  "decoder_layers": 6,
+  "decoder_normalize_before": true,
+  "decoder_retention_heads": 2,
+  "decoder_value_embed_dim": 864,
+  "deepnorm": false,
+  "drop_path_rate": 0.0,
+  "dropout": 0.0,
+  "eos_token_id": 0,
+  "forward_impl": "parallel",
+  "initializer_range": 0.02,
+  "is_decoder": true,
+  "layernorm_embedding": true,
+  "layernorm_eps": 1e-06,
+  "model_type": "retnet",
+  "no_scale_embedding": false,
+  "output_retentions": false,
+  "pad_token_id": 0,
+  "recurrent_chunk_size": 512,
+  "subln": true,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float16",
+  "transformers_version": "4.35.0.dev0",
+  "use_cache": true,
+  "use_ffn_rms_norm": false,
+  "use_glu": true,
+  "use_lm_decay": false,
+  "vocab_size": 65536,
+  "z_loss_coeff": 0.0
+}

generation_config.json

@@ -0,0 +1,6 @@
+{
+  "_from_model_config": true,
+  "eos_token_id": 0,
+  "pad_token_id": 0,
+  "transformers_version": "4.35.0.dev0"
+}

pytorch_model.bin

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