fix xformers?

This commit is contained in:
Drake Panzer 2022-10-28 14:30:22 -06:00
parent 4a65d9b9dc
commit a11f9020e4
1 changed files with 11 additions and 8 deletions

View File

@ -90,7 +90,7 @@
"activate_deepdanbooru = False # Enable and install DeepDanbooru -> https://github.com/KichangKim/DeepDanbooru\n",
"\n",
"# Don't put a trailing slash on directory paths.\n",
"# To reset your storage directory, rerun this cell.\n",
"# If you change any settings here, rerun this cell.\n",
"\n",
"# ===============================================================\n",
"# Save variables to Jupiter's temp storage so we can access it even if the kernel restarts.\n",
@ -279,19 +279,19 @@
" from subprocess import getoutput\n",
" s = getoutput('nvidia-smi')\n",
" if 'A4000' in s:\n",
" !pip install https://raw.githubusercontent.com/Cyberes/xformers-compiled/main/A4000/xformers-0.0.14.dev0-cp39-cp39-linux_x86_64.whl\n",
" !pip install --force-reinstall https://raw.githubusercontent.com/Cyberes/xformers-compiled/main/A4000/xformers-0.0.14.dev0-cp39-cp39-linux_x86_64.whl\n",
" elif 'P5000' in s:\n",
" !pip install https://raw.githubusercontent.com/Cyberes/xformers-compiled/main/P5000/xformers-0.0.14.dev0-cp39-cp39-linux_x86_64.whl\n",
" !pip install --force-reinstall https://raw.githubusercontent.com/Cyberes/xformers-compiled/main/P5000/xformers-0.0.14.dev0-cp39-cp39-linux_x86_64.whl\n",
" elif 'RTX 5000' in s:\n",
" !pip install https://raw.githubusercontent.com/Cyberes/xformers-compiled/main/RTX%205000/xformers-0.0.14.dev0-cp39-cp39-linux_x86_64.whl\n",
" !pip install --force-reinstall https://raw.githubusercontent.com/Cyberes/xformers-compiled/main/RTX%205000/xformers-0.0.14.dev0-cp39-cp39-linux_x86_64.whl\n",
" elif 'RTX 4000' in s:\n",
" !pip install https://raw.githubusercontent.com/Cyberes/xformers-compiled/main/RTX%204000/xformers-0.0.14.dev0-cp39-cp39-linux_x86_64.whl\n",
" !pip install --force-reinstall https://raw.githubusercontent.com/Cyberes/xformers-compiled/main/RTX%204000/xformers-0.0.14.dev0-cp39-cp39-linux_x86_64.whl\n",
" elif 'A100' in s:\n",
" !pip install https://raw.githubusercontent.com/Cyberes/xformers-compiled/main/A100/xformers-0.0.14.dev0-cp39-cp39-linux_x86_64.whl\n",
" !pip install --force-reinstall https://raw.githubusercontent.com/Cyberes/xformers-compiled/main/A100/xformers-0.0.14.dev0-cp39-cp39-linux_x86_64.whl\n",
" elif 'A6000' in s:\n",
" !pip install https://raw.githubusercontent.com/Cyberes/xformers-compiled/main/A6000/xformers-0.0.14.dev0-cp39-cp39-linux_x86_64.whl\n",
" !pip install --force-reinstall https://raw.githubusercontent.com/Cyberes/xformers-compiled/main/A6000/xformers-0.0.14.dev0-cp39-cp39-linux_x86_64.whl\n",
" else:\n",
" !pip install https://raw.githubusercontent.com/Cyberes/xformers-compiled/main/various/xformers-0.0.14.dev0-cp37-cp37m-linux_x86_64.whl\n",
" !pip install --force-reinstall https://raw.githubusercontent.com/Cyberes/xformers-compiled/main/various/xformers-0.0.14.dev0-cp37-cp37m-linux_x86_64.whl\n",
" print('GPU not matched to xformers binary so a one-size-fits-all binary was installed. If you have any issues, please build xformers using the Tools block below')\n",
" # print('XFORMERS ERROR: no wheel built for your GPU! Xformers disabled.')\n",
" # activate_xformers = False\n",
@ -724,6 +724,9 @@
" \n",
"def symlink_models(source_dir, filetype, destination_dir):\n",
" # don't pass directory paths with trailing slash\n",
" if not os.path.isdir(source_dir):\n",
" print('Not a directory:', source_dir)\n",
" return\n",
" for file in os.listdir(source_dir):\n",
" if file.endswith(filetype):\n",
" path = f'{destination_dir}/{file}'\n",