fix: add logger import to llama.py for unknown weight skip warning

The previous commit added a warning log for skipping unknown weights
(e.g. embed_tokens.biases) but missed importing the logger.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Chranos
2026-02-09 13:13:56 +08:00
parent 8efce7c44c
commit 6b708a43d8

View File

@@ -26,6 +26,10 @@ import torch
from torch import nn
from transformers import LlamaConfig
from vllm.logger import init_logger
logger = init_logger(__name__)
from vllm.attention import Attention, AttentionMetadata
from vllm.compilation.decorators import support_torch_compile
from vllm.config import CacheConfig, VllmConfig