[MISC] Skip dumping inputs when unpicklable (vllm-project#8744)

Signed-off-by: Amit Garg <mitgarg17495@gmail.com>

Author: comaniac

vllm/worker/model_runner_base.py

@@ -137,7 +137,15 @@ def _wrapper(*args, **kwargs):
                                                       for t in kv_caches
                                                       if is_tensor(t)]
 
-                    pickle.dump(dumped_inputs, filep)
+                    try:
+                        pickle.dump(dumped_inputs, filep)
+                    except Exception as pickle_err:
+                        logger.warning(
+                            "Failed to pickle inputs of failed execution: %s",
+                            str(pickle_err))
+                        raise type(err)(f"Error in model execution: "
+                                        f"{str(err)}") from err
+
                     logger.info(
                         "Completed writing input of failed execution to %s.",
                         filename)