Merge pull request #4879 from fracapuano/main

Add Single-Token predictions support for Replicate
This commit is contained in:
Krish Dholakia 2024-07-25 19:10:57 -07:00 committed by GitHub
commit c0c934d9cf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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