fix: fix init_llm.py

Author: ChenZiHong-Gavin

graphgen/operators/init/init_llm.py

@@ -12,12 +12,8 @@ class LLMFactory:
     - http_api: HTTPClient
     - openai_api: OpenAIClient
     - ollama_api: OllamaClient
-    - ollama: OllamaWrapper
-    - deepspeed: DeepSpeedWrapper
     - huggingface: HuggingFaceWrapper
-    - tgi: TGIWrapper
     - sglang: SGLangWrapper
-    - tensorrt: TensorRTWrapper
     """
 
     @staticmethod
@@ -47,6 +43,7 @@ def create_llm_wrapper(backend: str, config: Dict[str, Any]) -> BaseLLMWrapper:
             from graphgen.models.llm.local.sglang_wrapper import SGLangWrapper
 
             return SGLangWrapper(**config)
+
         # if backend == "vllm":
         #     from graphgen.models.llm.local.vllm_wrapper import VLLMWrapper
         #