ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-19 20:57:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

upgraded

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-11-19 20:23:00 +01:00
parent c7d6aab3f5
commit 1cb0fa02a4
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lollms/extension.py
View File

@ -28,6 +28,8 @@ class LOLLMSExtension():
self.app = app
self.config = config
self.script_path = script_path
self.category = str(script_path).replace("\\","/").split("/")[-2]
self.extension_folder_name = str(script_path).replace("\\","/").split("/")[-1]
self.card = self.script_path /"card.yaml"
self.configuration_path = app.lollms_paths.personal_configuration_path/"extensions"/f"{name}"

2
lollms/utilities.py
View File

@ -110,7 +110,7 @@ def check_torch_version(min_version, min_cuda_versio=12):
def reinstall_pytorch_with_cuda():
try:
ASCIIColors.info("Installing cuda 12.1.1")
result = subprocess.run(["conda", "install", "-c", "nvidia/label/cuda-12.1.1", "cuda-toolkit"])
result = subprocess.run(["conda", "install", "-c", "nvidia/label/cuda-12.1.1", "cuda-toolkit", "-y"])
except Exception as ex:
ASCIIColors.error(ex)
try: