diff --git a/lollms/binding.py b/lollms/binding.py index 82b0456..5724818 100644 --- a/lollms/binding.py +++ b/lollms/binding.py @@ -19,6 +19,7 @@ from tqdm import tqdm import urllib.request import importlib import shutil +import subprocess __author__ = "parisneo" @@ -203,6 +204,16 @@ class BindingInstaller: def __init__(self, config: LOLLMSConfig) -> None: self.config = config + def reinstall_pytorch_with_cuda(self): + result = subprocess.run(["pip", "install", "--upgrade", "torch", "torchvision", "torchaudio", "--no-cache-dir", "--index-url", "https://download.pytorch.org/whl/cu117"]) + if result.returncode != 0: + ASCIIColors.warning("Couldn't find Cuda build tools on your PC. Reverting to CPU.") + result = subprocess.run(["pip", "install", "--upgrade", "torch", "torchvision", "torchaudio", "--no-cache-dir"]) + if result.returncode != 0: + ASCIIColors.error("Couldn't install pytorch !!") + else: + ASCIIColors.error("Pytorch installed successfully!!") + class LLMBinding: diff --git a/lollms/bindings_zoo b/lollms/bindings_zoo index 9c46584..6ccb393 160000 --- a/lollms/bindings_zoo +++ b/lollms/bindings_zoo @@ -1 +1 @@ -Subproject commit 9c4658498e2a42627d6d6cb113798e8d9fcff3b3 +Subproject commit 6ccb39333feec960d20846dd96ef3df798697bc5 diff --git a/lollms/personalities_zoo b/lollms/personalities_zoo index 259ab42..442b890 160000 --- a/lollms/personalities_zoo +++ b/lollms/personalities_zoo @@ -1 +1 @@ -Subproject commit 259ab420e909bf4aab935c053c1b3aeed2fee719 +Subproject commit 442b890763f66833d1df0940fa8f2dc3572f833d diff --git a/lollms/personality.py b/lollms/personality.py index a26f315..ffac2ed 100644 --- a/lollms/personality.py +++ b/lollms/personality.py @@ -257,7 +257,6 @@ def install_package(package_name): print(f"{package_name} has been successfully installed.") - class AIPersonality: # Extra @@ -1041,3 +1040,6 @@ Date: {{date}} output_string = re.sub(pattern, replace, input_string) return output_string +class AIPersonalityInstaller: + def __init__(self, personality:AIPersonality) -> None: + self.personality = personality