deskull commited on
Commit
85ffaab
·
verified ·
1 Parent(s): 0a07cad

Update model card: restore eos_token_id=None workaround, fix BERT mask_token, correct RNA prompt

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -19,7 +19,7 @@ GPT-2 medium (345M parameters) foundation model pre-trained on protein amino aci
19
 
20
  - **Model Type**: gpt2
21
  - **Data Type**: Protein
22
- - **Training Date**: 2026-04-13
23
 
24
  ## Usage
25
 
@@ -39,7 +39,7 @@ with torch.no_grad():
39
  max_new_tokens=50,
40
  do_sample=True,
41
  temperature=0.8,
42
- eos_token_id=tokenizer.eos_token_id,
43
  pad_token_id=0,
44
  )
45
  print(tokenizer.decode(output_ids[0], skip_special_tokens=True))
 
19
 
20
  - **Model Type**: gpt2
21
  - **Data Type**: Protein
22
+ - **Training Date**: 2026-04-14
23
 
24
  ## Usage
25
 
 
39
  max_new_tokens=50,
40
  do_sample=True,
41
  temperature=0.8,
42
+ eos_token_id=None, # HF config.json has legacy eos_token_id=0; disable early stop
43
  pad_token_id=0,
44
  )
45
  print(tokenizer.decode(output_ids[0], skip_special_tokens=True))