1) apply multi stage build, 2) rm gemini
This commit is contained in:
@ -36,13 +36,6 @@ class TranslatorApp:
|
||||
keep_alive=self._config['app']['ollama']['keep_alive'],
|
||||
stop=None
|
||||
)
|
||||
elif self._config['app']['llm_mode'] == 'gemini':
|
||||
llm = ChatGoogleGenerativeAI(
|
||||
model=self._config['app']['gemini']['model_name'],
|
||||
temperature=self._config['app']['gemini']['temperature'],
|
||||
max_tokens=self._config['app']['gemini']['max_tokens'],
|
||||
top_p=self._config['app']['gemini']['top_p']
|
||||
)
|
||||
elif self._config['app']['llm_mode'] == 'openai':
|
||||
llm = ChatOpenAI(
|
||||
model=self._config['app']['openai']['model_name'],
|
||||
|
||||
Reference in New Issue
Block a user