KapitalK commited on
Commit
bed490c
1 Parent(s): d709e04

Upload model

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. adapter_config.json +3 -3
  3. adapter_model.bin +2 -2
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  library_name: peft
3
- base_model: openai-community/gpt2
4
  ---
5
 
6
  # Model Card for Model ID
 
1
  ---
2
  library_name: peft
3
+ base_model: EleutherAI/gpt-neo-125m
4
  ---
5
 
6
  # Model Card for Model ID
adapter_config.json CHANGED
@@ -1,11 +1,11 @@
1
  {
2
  "auto_mapping": null,
3
- "base_model_name_or_path": "openai-community/gpt2",
4
  "inference_mode": true,
5
  "num_attention_heads": 12,
6
  "num_layers": 12,
7
  "num_transformer_submodules": 1,
8
- "num_virtual_tokens": 2,
9
  "peft_type": "PROMPT_TUNING",
10
  "prompt_tuning_init": "TEXT",
11
  "prompt_tuning_init_text": "Classify if the tweet is a complaint or not",
@@ -13,5 +13,5 @@
13
  "task_type": "CAUSAL_LM",
14
  "token_dim": 768,
15
  "tokenizer_kwargs": null,
16
- "tokenizer_name_or_path": "openai-community/gpt2"
17
  }
 
1
  {
2
  "auto_mapping": null,
3
+ "base_model_name_or_path": "EleutherAI/gpt-neo-125m",
4
  "inference_mode": true,
5
  "num_attention_heads": 12,
6
  "num_layers": 12,
7
  "num_transformer_submodules": 1,
8
+ "num_virtual_tokens": 8,
9
  "peft_type": "PROMPT_TUNING",
10
  "prompt_tuning_init": "TEXT",
11
  "prompt_tuning_init_text": "Classify if the tweet is a complaint or not",
 
13
  "task_type": "CAUSAL_LM",
14
  "token_dim": 768,
15
  "tokenizer_kwargs": null,
16
+ "tokenizer_name_or_path": "EleutherAI/gpt-neo-125m"
17
  }
adapter_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a6fc5b40508be06011dfc0cd420ff4bac5f71148152ea3429c3b3c6c56fc11a1
3
- size 7418
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c48027da02f5ddce344af1424e630b26e2c2901907cf36b0f927750353588a1a
3
+ size 25850