Hugging Face's logo

ZhankuiHe
/
gpt8gen

Transformers
PyTorch
switch4gen
Inference Endpoints
Model card Files Community
ZhankuiHe commited on
Commit
8d16928
•
1 Parent(s): 283a851

Upload Switch4GenModule

Browse files
Files changed (3) hide show
  1. config.json +5 -1
  2. generation_config.json +1 -1
  3. pytorch_model.bin +1 -1
config.json CHANGED
@@ -5,7 +5,11 @@
5
  "dialog_model": "gpt2",
6
  "model_type": "switch4gen",
7
  "num_hidden_layers": 4,
 
 
 
 
8
  "switch_token_id": 50257,
9
  "torch_dtype": "float32",
10
- "transformers_version": "4.40.0"
11
  }
 
5
  "dialog_model": "gpt2",
6
  "model_type": "switch4gen",
7
  "num_hidden_layers": 4,
8
+ "stop_tokens": [
9
+ 50256,
10
+ 50257
11
+ ],
12
  "switch_token_id": 50257,
13
  "torch_dtype": "float32",
14
+ "transformers_version": "4.44.0"
15
  }
generation_config.json CHANGED
@@ -1,4 +1,4 @@
1
  {
2
  "_from_model_config": true,
3
- "transformers_version": "4.40.0"
4
  }
 
1
  {
2
  "_from_model_config": true,
3
+ "transformers_version": "4.44.0"
4
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3c53249e19d49d33a95e28e2701a67de62a408e0217e3d256baa0b9824b13657
3
  size 497817378
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:33aeae6ab546de1e0afac666a73b70d337d7562aa58e2f44b8c5645ce7defe21
3
  size 497817378