fix: ollama custom model

This commit is contained in:
Timothy Jaeryang Baek 2024-11-17 16:01:47 -08:00
parent 3fa23481c8
commit 6589f75b2d

View File

@ -931,9 +931,6 @@ async def generate_chat_completion(
del payload["metadata"]
model_id = payload["model"]
if ":" not in model_id:
model_id = f"{model_id}:latest"
model_info = Models.get_model_by_id(model_id)
if model_info: