xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-22 06:35:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[misc] use tqdm.auto where appropriate (#16290)

Browse Source 
Signed-off-by: Benjamin Kitor <bkitor@gigaio.com>
This commit is contained in:
Benjamin Kitor 2025-04-09 21:54:54 -07:00 committed by GitHub
parent 0d4d06fe2f
commit 82eb61dd4c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/entrypoints/llm.py
View File

@ -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,

2
vllm/worker/model_runner.py
View File

@ -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