Skip to content

Commit b182329

Browse files
committed
[Bugfix][LoRA][Operator] fix lint
Signed-off-by: paulyu12 <507435917@qq.com>
1 parent 50e8de8 commit b182329

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

vllm_ascend/patch/worker/patch_common/patch_lora_embedding.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212

1313

1414
class AscendVocabParallelEmbeddingWithLoRA(VocabParallelEmbeddingWithLoRA):
15+
1516
@classmethod
1617
def can_replace_layer(
1718
cls,
@@ -22,6 +23,7 @@ def can_replace_layer(
2223
) -> bool:
2324
return type(source_layer) is AscendVocabParallelEmbedding
2425

26+
2527
# Patch for lora register_model issue after overriding VocabParallelEmbedding class (#2515)
2628
_all_lora_classes.add(AscendVocabParallelEmbeddingWithLoRA)
27-
vllm.lora.utils._all_lora_classes = _all_lora_classes
29+
vllm.lora.utils._all_lora_classes = _all_lora_classes

0 commit comments

Comments
 (0)