ZhankuiHe commited on
Commit
ff45201
·
verified ·
1 Parent(s): 3928b6a

Upload Switch4GenModule

Browse files
Files changed (2) hide show
  1. config.json +1 -1
  2. pytorch_model.bin +1 -1
config.json CHANGED
@@ -2,7 +2,7 @@
2
  "architectures": [
3
  "Switch4GenModule"
4
  ],
5
- "dialog_model": "gpt2",
6
  "model_type": "switch4gen",
7
  "num_hidden_layers": 4,
8
  "switch_token_id": 50257,
 
2
  "architectures": [
3
  "Switch4GenModule"
4
  ],
5
+ "dialog_model": "gpt2-tiny",
6
  "model_type": "switch4gen",
7
  "num_hidden_layers": 4,
8
  "switch_token_id": 50257,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:827677806e90c82ef0dbf3cd86ae8150d48db027b9105af6a6a5e27f6db07b97
3
  size 270974288
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1e123be660d6693397da6f86b8e14c1f026e2ad40a9ece9add9f0f7aad04275c
3
  size 270974288