fix: add tp_size for sglang_wrapper

Author: ChenZiHong-Gavin

graphgen/models/llm/local/sglang_wrapper.py

@@ -16,6 +16,7 @@ def __init__(
         temperature: float = 0.0,
         top_p: float = 1.0,
         topk: int = 5,
+        tp_size: int = 1,
         **kwargs: Any,
     ):
         super().__init__(temperature=temperature, top_p=top_p, **kwargs)
@@ -34,7 +35,7 @@ def __init__(
         self.topk = topk
 
         # Initialise the offline engine
-        self.engine = sgl.Engine(model_path=self.model_path)
+        self.engine = sgl.Engine(model_path=self.model_path, tp_size=int(tp_size))
 
         # Keep helpers for streaming
         self.async_stream_and_merge = async_stream_and_merge