20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

support tqdm in notebooks (#7510)

Browse Source
This commit is contained in:
fzyzcjy 2024-08-17 00:17:50 +08:00 committed by GitHub
parent 0e39a33c6d
commit ec724a725e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/entrypoints/llm.py
View File

@ -1,7 +1,7 @@
from contextlib import contextmanager from contextlib import contextmanager
from typing import ClassVar, List, Optional, Sequence, Union, cast, overload from typing import ClassVar, List, Optional, Sequence, Union, cast, overload
from tqdm import tqdm from tqdm.auto import tqdm
from transformers import PreTrainedTokenizer, PreTrainedTokenizerFast from transformers import PreTrainedTokenizer, PreTrainedTokenizerFast
from vllm.engine.arg_utils import EngineArgs from vllm.engine.arg_utils import EngineArgs