From 71aae3aba45af4a91ba9fa22c685f9ff6d7fb08b Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Mon, 13 Jan 2025 15:33:43 +0900 Subject: [PATCH] fix zluda installer bug --- cli/zluda-python.py | 7 +++---- modules/zluda_installer.py | 6 +++--- 2 files changed, 6 insertions(+), 7 deletions(-) diff --git a/cli/zluda-python.py b/cli/zluda-python.py index e0399d096..894489b74 100644 --- a/cli/zluda-python.py +++ b/cli/zluda-python.py @@ -27,10 +27,9 @@ if __name__ == '__main__': sys.path.append(os.getcwd()) from modules import zluda_installer - zluda_path = zluda_installer.get_path() - zluda_installer.install(zluda_path) - zluda_installer.make_copy(zluda_path) - zluda_installer.load(zluda_path) + zluda_installer.install() + zluda_installer.make_copy() + zluda_installer.load() import torch interpreter = Interpreter({ diff --git a/modules/zluda_installer.py b/modules/zluda_installer.py index 6c74ee6b8..b49434aeb 100644 --- a/modules/zluda_installer.py +++ b/modules/zluda_installer.py @@ -41,13 +41,13 @@ def install() -> None: for info in infos: if not info.is_dir(): info.filename = os.path.basename(info.filename) - archive.extract(info, '.zluda') + archive.extract(info, path) os.remove('_zluda') def uninstall() -> None: - if os.path.exists('.zluda'): - shutil.rmtree('.zluda') + if os.path.exists(path): + shutil.rmtree(path) def set_blaslt_enabled(enabled: bool):