fix: now supports single tokens prediction

2024-07-25 19:06:07 +02:00 · 2024-07-25 19:06:07 +02:00 · 5553f84d51
commit 5553f84d51
parent 80800b9ec8
1 changed files with 1 additions and 1 deletions
--- a/litellm/llms/replicate.py
+++ b/litellm/llms/replicate.py
@ -387,7 +387,7 @@ def process_response(
        result = " "

    ## Building RESPONSE OBJECT
-    if len(result) > 1:
+    if len(result) >= 1:
        model_response.choices[0].message.content = result  # type: ignore

    # Calculate usage