Skip to content

Commit

Permalink
fix set flag to raw
Browse files Browse the repository at this point in the history
  • Loading branch information
davidberenstein1957 committed Dec 19, 2024
1 parent c6708c9 commit 2ddfa3f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/distilabel/models/llms/ollama.py
Original file line number Diff line number Diff line change
Expand Up @@ -239,6 +239,7 @@ async def _generate_with_text_generation(
format=format,
options=options,
keep_alive=keep_alive,
raw=True,
)

@validate_call
Expand Down Expand Up @@ -277,7 +278,6 @@ async def agenerate(
input, format, options, keep_alive
)
)

text = completion.response
except Exception as e:
self._logger.warning( # type: ignore
Expand Down

0 comments on commit 2ddfa3f

Please sign in to comment.