NoteBook-renamed / memfix.py
imspidey's picture
Create memfix.py
8a24a8c
raw
history blame contribute delete
557 Bytes
import subprocess
import os
import zipfile
memfix_url = "https://huggingface.co/imspidey/NoteBook/resolve/main/memfix.zip"
memfix_zip_path = "/content/memfix.zip"
subprocess.run(["curl", "-Lo", memfix_zip_path, memfix_url])
unzip_dir = "/content"
os.makedirs(unzip_dir, exist_ok=True)
with zipfile.ZipFile(memfix_zip_path, 'r') as zip_ref:
zip_ref.extractall(unzip_dir)
subprocess.run(["apt", "-y", "update", "-qq"])
os.environ["LD_PRELOAD"] = "/content/memfix/libtcmalloc_minimal.so.4"
print("memfix.zip downloaded, extracted, and LD_PRELOAD set.")