fix: fix init_llm.py

Author: ChenZiHong-Gavin

graphgen/operators/init/init_llm.py

@@ -43,15 +43,14 @@ def create_llm_wrapper(backend: str, config: Dict[str, Any]) -> BaseLLMWrapper:
             from graphgen.models.llm.local.hf_wrapper import HuggingFaceWrapper
 
             return HuggingFaceWrapper(**config)
-        # if backend == "sglang":
-        #     from graphgen.models.llm.local.sglang_wrapper import SGLangWrapper
-        #
-        #     return SGLangWrapper(**config)
-
-        if backend == "vllm":
-            from graphgen.models.llm.local.vllm_wrapper import VLLMWrapper
+        if backend == "sglang":
+            from graphgen.models.llm.local.sglang_wrapper import SGLangWrapper
 
-            return VLLMWrapper(**config)
+            return SGLangWrapper(**config)
+        # if backend == "vllm":
+        #     from graphgen.models.llm.local.vllm_wrapper import VLLMWrapper
+        #
+        #     return VLLMWrapper(**config)
 
         raise NotImplementedError(f"Backend {backend} is not implemented yet.")