Update model card: restore eos_token_id=None workaround, fix BERT mask_token, correct RNA prompt
Browse files
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-
|
| 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=
|
| 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))
|