From 82eb61dd4c4e306bda4f20edab063693396c4e1a Mon Sep 17 00:00:00 2001 From: Benjamin Kitor Date: Wed, 9 Apr 2025 21:54:54 -0700 Subject: [PATCH] [misc] use tqdm.auto where appropriate (#16290) Signed-off-by: Benjamin Kitor --- vllm/entrypoints/llm.py | 2 +- vllm/worker/model_runner.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/vllm/entrypoints/llm.py b/vllm/entrypoints/llm.py index c79ab16a..70bb73f4 100644 --- a/vllm/entrypoints/llm.py +++ b/vllm/entrypoints/llm.py @@ -8,7 +8,7 @@ from typing import Any, Callable, ClassVar, Optional, Union, cast, overload import cloudpickle import torch.nn as nn -from tqdm import tqdm +from tqdm.auto import tqdm from typing_extensions import TypeVar, deprecated from vllm.beam_search import (BeamSearchInstance, BeamSearchOutput, diff --git a/vllm/worker/model_runner.py b/vllm/worker/model_runner.py index 86e6d975..9524a69f 100644 --- a/vllm/worker/model_runner.py +++ b/vllm/worker/model_runner.py @@ -15,7 +15,7 @@ import numpy as np import torch import torch.distributed import torch.nn as nn -from tqdm import tqdm +from tqdm.auto import tqdm import vllm.envs as envs from vllm.attention import AttentionMetadata, get_attn_backend