Skip to content

Commit

Permalink
avoid pylint error by using direct import
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 673833952
Change-Id: I087e4d8de1d81b8bedb89f1b886ffe3a87076374
  • Loading branch information
jagapiou authored and copybara-github committed Sep 12, 2024
1 parent 5c2bb9e commit 64c9e56
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions concordia/language_model/langchain_ollama_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
from concordia.language_model import language_model
from concordia.utils import measurements as measurements_lib
from concordia.utils import sampling
from langchain import llms
from langchain.llms import ollama

from typing_extensions import override

Expand Down Expand Up @@ -64,7 +64,7 @@ def __init__(
self._terminators = []
if 'llama3' in self._model_name:
self._terminators.extend(['<|eot_id|>'])
self._client = llms.Ollama(model=model_name, stop=self._terminators)
self._client = ollama.Ollama(model=model_name, stop=self._terminators)

self._measurements = measurements
self._channel = channel
Expand Down

0 comments on commit 64c9e56

Please sign in to comment.