[BugFix] fix wrong output when using lora and num_scheduler_steps=8 (#11161)

FIX issue #9688
#11086 #12487

---------

Signed-off-by: Jee Jee Li <[email protected]>
Co-authored-by: weilong.yu <[email protected]>
Co-authored-by: Jee Jee Li <[email protected]>

Author: 3 people

vllm/worker/model_runner.py

@@ -1346,6 +1346,10 @@ def _dummy_run(self,
 
             self.execute_model(model_input, kv_caches, intermediate_tensors)
             torch.cuda.synchronize()
+            if self.lora_config:
+                # Remove dummy loras.
+                assert self.lora_manager is not None
+                self.remove_all_loras()
             return
 
     def remove_all_loras(self):

vllm/worker/worker.py

@@ -264,10 +264,7 @@ def determine_num_available_blocks(self) -> Tuple[int, int]:
                f"{(available_kv_cache_memory / GiB_bytes):.2f}GiB.")
 
         logger.info(msg)
-
         # Final cleanup
-        if self.model_runner.lora_manager:
-            self.model_runner.remove_all_loras()
         gc.collect()
 
         return num_gpu_blocks, num_cpu_blocks