Hafeez7000 commited on
Commit
cf8e9df
·
verified ·
1 Parent(s): 1b0be3b

Upload model

Browse files
Files changed (2) hide show
  1. config.json +4 -3
  2. model.safetensors +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "talkai-small",
3
  "activation_function": "gelu_new",
4
  "architectures": [
5
  "GPT2LMHeadModel"
@@ -27,8 +27,9 @@
27
  "summary_type": "cls_index",
28
  "summary_use_proj": true,
29
  "task_specific_params": {
30
- "conversational": {
31
- "max_length": 1000
 
32
  }
33
  },
34
  "torch_dtype": "float32",
 
1
  {
2
+ "_name_or_path": "gpt2",
3
  "activation_function": "gelu_new",
4
  "architectures": [
5
  "GPT2LMHeadModel"
 
27
  "summary_type": "cls_index",
28
  "summary_use_proj": true,
29
  "task_specific_params": {
30
+ "text-generation": {
31
+ "do_sample": true,
32
+ "max_length": 50
33
  }
34
  },
35
  "torch_dtype": "float32",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ed4352057840e897e31076ccbfc9c857b14cc1ecd6dd9a4b7b5ee2822b560cdd
3
  size 497774208
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7d00560d8910fbed77ffad4065dee5011c41ba401b1064e749c498ba9e20373
3
  size 497774208