Spaces:
Running
Running
File size: 550 Bytes
854b7af 0558cbb 412c787 854b7af 0558cbb 412c787 3a236ec 40874a6 412c787 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import psutil
from huggingface_hub import Repository
import os
def get_current_ram_usage():
ram = psutil.virtual_memory()
return ram.available / 1024 / 1024 / 1024, ram.total / 1024 / 1024 / 1024
def download_models(models):
model_dirs = {}
for model in models:
model_dirs[model] = Repository(
model, clone_from=f"https://huggingface.co/researchaccount/{model}"
)
return model_dirs
def install_git_lfs():
os.system("chmod -R 777 /home/user/app/.git/hooks/")
os.system("git lfs install")
|