xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-06-08 11:29:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix pre-commit

Browse Source 
Signed-off-by: bk-201 <joy25810@foxmail.com>
This commit is contained in:
bk-201 2025-11-25 14:45:49 +00:00
parent ee0cdef382
commit a3647878c2
2 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
vllm/lora/models.py
View File

@ -13,7 +13,12 @@ from torch import nn
from vllm.config.lora import LoRAConfig, ModelConfig from vllm.config.lora import LoRAConfig, ModelConfig
from vllm.logger import init_logger from vllm.logger import init_logger
from vllm.lora.layers import BaseLayerWithLoRA, LoRAMapping, LoRAMappingType from vllm.lora.layers import (
BaseLayerWithLoRA,
FusedMoEWithLoRA,
LoRAMapping,
LoRAMappingType,
)
from vllm.lora.lora_weights import LoRALayerWeights, PackedLoRALayerWeights from vllm.lora.lora_weights import LoRALayerWeights, PackedLoRALayerWeights
from vllm.lora.peft_helper import PEFTHelper from vllm.lora.peft_helper import PEFTHelper
from vllm.lora.punica_wrapper import PunicaWrapperBase, get_punica_wrapper from vllm.lora.punica_wrapper import PunicaWrapperBase, get_punica_wrapper
@ -25,7 +30,6 @@ from vllm.lora.utils import (
is_moe_model, is_moe_model,
is_regex_target_modules, is_regex_target_modules,
parse_fine_tuned_lora_name, parse_fine_tuned_lora_name,
process_packed_modules_mapping,
replace_submodule, replace_submodule,
) )
from vllm.model_executor.layers.fused_moe import FusedMoE from vllm.model_executor.layers.fused_moe import FusedMoE
@ -36,7 +40,6 @@ from vllm.model_executor.models.module_mapping import MultiModelKeys
from vllm.model_executor.models.utils import PPMissingLayer, WeightsMapper from vllm.model_executor.models.utils import PPMissingLayer, WeightsMapper
from vllm.model_executor.utils import get_packed_modules_mapping from vllm.model_executor.utils import get_packed_modules_mapping
from vllm.multimodal import MULTIMODAL_REGISTRY from vllm.multimodal import MULTIMODAL_REGISTRY
from vllm.utils import is_pin_memory_available
from vllm.utils.cache import LRUCache from vllm.utils.cache import LRUCache
from vllm.utils.platform_utils import is_pin_memory_available from vllm.utils.platform_utils import is_pin_memory_available

8
vllm/v1/worker/lora_model_runner_mixin.py
View File

@ -178,7 +178,7 @@ class LoRAModelRunnerMixin:
tuple(sample_lora_mapping), tuple(sample_lora_mapping),
tuple(token_lora_mapping), tuple(token_lora_mapping),
lora_requests, lora_requests,
mapping_type mapping_type,
) )
yield yield
@ -196,7 +196,11 @@ class LoRAModelRunnerMixin:
with ( with (
self.maybe_setup_dummy_loras(lora_config, remove_lora), self.maybe_setup_dummy_loras(lora_config, remove_lora),
self.maybe_select_dummy_loras( self.maybe_select_dummy_loras(
lora_config, num_scheduled_tokens, mapping_type, num_sampled_tokens, activate_lora lora_config,
num_scheduled_tokens,
mapping_type,
num_sampled_tokens,
activate_lora,
), ),
): ):
yield yield