super-dainiu commited on
Commit
1db1c65
·
verified ·
1 Parent(s): d8ccf56

Upload config

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -1,6 +1,7 @@
1
  {
2
- "_name_or_path": "Yale-CompBio/tcrlang-hf",
3
  "architectures": [
 
4
  "AbLang2ForMaskedLM"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
@@ -18,7 +19,6 @@
18
  "pad_token_id": 0,
19
  "qk_layernorm": false,
20
  "residue_scaling": 1.0,
21
- "torch_dtype": "float32",
22
  "transformers_version": "4.46.3",
23
  "use_flash_attn": false,
24
  "vocab_size": 26
 
1
  {
2
+ "_attn_implementation_autoset": true,
3
  "architectures": [
4
+ "AbLang2Model",
5
  "AbLang2ForMaskedLM"
6
  ],
7
  "attention_probs_dropout_prob": 0.1,
 
19
  "pad_token_id": 0,
20
  "qk_layernorm": false,
21
  "residue_scaling": 1.0,
 
22
  "transformers_version": "4.46.3",
23
  "use_flash_attn": false,
24
  "vocab_size": 26