Skip to content

Commit f475f9e

Browse files
committed
Add test case
1 parent afff715 commit f475f9e

File tree

3 files changed

+9
-0
lines changed

3 files changed

+9
-0
lines changed

site/docs/supported-models/_components/llm-models-table/models.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -195,6 +195,7 @@ export const LLM_MODELS: LLMModelType[] = [
195195
{
196196
name: 'Exaone',
197197
links: [
198+
'https://huggingface.co/LGAI-EXAONE/EXAONE-4.0-1.2B',
198199
'https://huggingface.co/LGAI-EXAONE/EXAONE-3.5-2.4B-Instruct',
199200
'https://huggingface.co/LGAI-EXAONE/EXAONE-3.5-7.8B-Instruct',
200201
'https://huggingface.co/LGAI-EXAONE/EXAONE-3.5-32B-Instruct',

tests/python_tests/data/tokenizer_configs.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -165,6 +165,13 @@ def get_tokenizer_configs():
165165
"unk_token": "[UNK]",
166166
"chat_template": "{% for message in messages %}{% if loop.first and message['role'] != 'system' %}{{ '[|system|][|endofturn|]\n' }}{% endif %}{{ '[|' + message['role'] + '|]' + message['content'] }}{% if message['role'] == 'user' %}{{ '\n' }}{% else %}{{ '[|endofturn|]\n' }}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '[|assistant|]' }}{% endif %}"
167167
},
168+
"LGAI-EXAONE/EXAONE-4.0-1.2B": {
169+
"bos_token": "[BOS]",
170+
"eos_token": "[|endofturn|]",
171+
"pad_token": "[PAD]",
172+
"unk_token": "[UNK]",
173+
"chat_template": "{% for message in messages %}{% if loop.first and message['role'] != 'system' %}{{ '[|system|][|endofturn|]\n' }}{% endif %}{{ '[|' + message['role'] + '|]' + message['content'] }}{% if message['role'] == 'user' %}{{ '\n' }}{% else %}{{ '[|endofturn|]\n' }}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '[|assistant|]' }}{% endif %}"
174+
},
168175
"meta-llama/Llama-2-70b-chat-hf": {
169176
"bos_token": "<s>",
170177
"eos_token": "</s>",

tests/python_tests/models/real_models

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ BAAI/AquilaChat2-7B
55
CohereForAI/c4ai-command-r-v01
66
EleutherAI/gpt-j-6B
77
EleutherAI/gpt-j-6b
8+
LGAI-EXAONE/EXAONE-4.0-1.2B
89
EleutherAI/gpt-neo-1.3B
910
EleutherAI/gpt-neo-125m
1011
EleutherAI/gpt-neo-2.7B

0 commit comments

Comments
 (0)