Skip to content

Commit

Permalink
fix UT but not aligned with HF
Browse files Browse the repository at this point in the history
  • Loading branch information
AllentDan committed Jan 21, 2025
1 parent 74a3aa4 commit 34850f7
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions tests/test_lmdeploy/test_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -632,12 +632,6 @@ def test_deepseek():
}, {
'role': 'user',
'content': 'who are you'
}, {
'role': 'assistant',
'content': 'I am an AI'
}, {
'role': 'user',
'content': 'hi'
}]
from transformers import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained('deepseek-ai/DeepSeek-V2-Lite',
Expand Down

0 comments on commit 34850f7

Please sign in to comment.