addressing comments

This commit is contained in:
Hardik Shah 2024-07-31 22:07:45 -07:00
parent 0e985648f5
commit fd8adc1e50
3 changed files with 9 additions and 2 deletions

View file

@ -13,7 +13,7 @@ async def get_inference_api_instance(config: InferenceConfig):
return InferenceImpl(config.impl_config)
elif config.impl_config.impl_type == ImplType.ollama.value:
from .inference import OllamaInference
from .ollama import OllamaInference
return OllamaInference(config.impl_config)