support tqdm in notebooks (#7510)

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

View File

@ -1,7 +1,7 @@
from contextlib import contextmanager
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 vllm.engine.arg_utils import EngineArgs