From 35a00a1514721849ca2c421b901de7e667470f1c Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Sun, 10 Dec 2023 22:24:03 +0100 Subject: [PATCH] Upgraded binding --- lollms/app.py | 2 +- lollms/apps/server/__init__.py | 2 +- lollms/apps/settings/__init__.py | 6 +++--- lollms/terminal.py | 2 +- 4 files changed, 6 insertions(+), 6 deletions(-) diff --git a/lollms/app.py b/lollms/app.py index e9ec005..bcd1894 100644 --- a/lollms/app.py +++ b/lollms/app.py @@ -308,7 +308,7 @@ class LollmsApplication(LoLLMsCom): self.info("Trying to reinstall binding") trace_exception(ex) try: - binding = BindingBuilder().build_binding(self.config, self.lollms_paths,installation_option=InstallOption.FORCE_INSTALL, app=self) + binding = BindingBuilder().build_binding(self.config, self.lollms_paths,installation_option=InstallOption.FORCE_INSTALL, lollmsCom=self) except Exception as ex: self.error("Couldn't reinstall binding") trace_exception(ex) diff --git a/lollms/apps/server/__init__.py b/lollms/apps/server/__init__.py index 0987c69..238538a 100644 --- a/lollms/apps/server/__init__.py +++ b/lollms/apps/server/__init__.py @@ -215,7 +215,7 @@ class LoLLMsServer(LollmsApplication): for per in self.mounted_personalities: per.model = None gc.collect() - self.binding = BindingBuilder().build_binding(self.config, self.lollms_paths, app=self) + self.binding = BindingBuilder().build_binding(self.config, self.lollms_paths, lollmsCom=self) self.model = self.binding.build_model() for per in self.mounted_personalities: per.model = self.model diff --git a/lollms/apps/settings/__init__.py b/lollms/apps/settings/__init__.py index 0869543..e86978e 100644 --- a/lollms/apps/settings/__init__.py +++ b/lollms/apps/settings/__init__.py @@ -226,13 +226,13 @@ def main(): if args.install_binding: settings_app.config.binding_name= args.install_binding - settings_app.binding = BindingBuilder().build_binding(settings_app.config, settings_app.lollms_paths,InstallOption.FORCE_INSTALL, app=settings_app) + settings_app.binding = BindingBuilder().build_binding(settings_app.config, settings_app.lollms_paths,InstallOption.FORCE_INSTALL, lollmsCom=settings_app) settings_app.config.save_config() if args.install_model: if not settings_app.binding: - settings_app.binding = BindingBuilder().build_binding(settings_app.config, settings_app.lollms_paths,InstallOption.FORCE_INSTALL, app=settings_app) + settings_app.binding = BindingBuilder().build_binding(settings_app.config, settings_app.lollms_paths,InstallOption.FORCE_INSTALL, lollmsCom=settings_app) def progress_callback(blocks, block_size, total_size): tqdm_bar.total=total_size @@ -249,7 +249,7 @@ def main(): if args.select_model: settings_app.config.model_name = args.select_model if not settings_app.binding: - settings_app.binding = BindingBuilder().build_binding(settings_app.config, settings_app.lollms_paths,InstallOption.FORCE_INSTALL, app=settings_app) + settings_app.binding = BindingBuilder().build_binding(settings_app.config, settings_app.lollms_paths,InstallOption.FORCE_INSTALL, lollmsCom=settings_app) settings_app.model = settings_app.binding.build_model() settings_app.config.save_config() diff --git a/lollms/terminal.py b/lollms/terminal.py index 5a055d3..22d519e 100644 --- a/lollms/terminal.py +++ b/lollms/terminal.py @@ -425,7 +425,7 @@ class MainMenu(Menu): self.lollms_app.config['binding_name']=self.binding_infs[choice - 1]['name'] try: - lollms_app.binding = BindingBuilder().build_binding(lollms_app.config, lollms_app.lollms_paths,InstallOption.FORCE_INSTALL, app=self) + lollms_app.binding = BindingBuilder().build_binding(lollms_app.config, lollms_app.lollms_paths,InstallOption.FORCE_INSTALL, lollmsCom=self) except Exception as ex: print(ex) print(f"Couldn't find binding. Please verify your configuration file at {lollms_app.config.file_path} or use the next menu to select a valid binding")