Fix test

Signed-off-by: Jee Jee Li <[email protected]>

Author: jeejeelee

tests/lora/test_llama_tp.py

@@ -11,6 +11,8 @@
 from vllm.lora.request import LoRARequest
 from vllm.model_executor.model_loader.tensorizer import TensorizerConfig
 
+from ..utils import VLLM_PATH, create_new_process_for_each_test
+
 PROMPT_TEMPLATE = """<|eot_id|><|start_header_id|>user<|end_header_id|>
 I want you to act as a SQL terminal in front of an example database, you need only to return the sql command to me.Below is an instruction that describes a task, Write a response that appropriately completes the request.
 "
@@ -113,6 +115,7 @@ def generate_and_test(
     print("removing lora")
 
 
+@create_new_process_for_each_test()
 @pytest.mark.parametrize("cudagraph_specialize_lora", [True, False])
 def test_llama_lora(llama32_lora_files, cudagraph_specialize_lora: bool):
     llm = vllm.LLM(

vllm/lora/utils.py

@@ -171,10 +171,9 @@ def is_base_embeddding_weights(name: str) -> bool:
     input_embedding_subfix = ".embed_tokens.base_layer.weight"
     output_embedding_subfix = ".lm_head.base_layer.weight"
 
-    if name.endswith(input_embedding_subfix) or name.endswith(output_embedding_subfix):
-        return True
-
-    return False
+    return name.endswith(input_embedding_subfix) or name.endswith(
+        output_embedding_subfix
+    )
 
 
 def is_regex_target_modules(