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):