fix xformers?
This commit is contained in:
parent
4a65d9b9dc
commit
a11f9020e4
|
@ -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",
|
||||
|
|
Loading…
Reference in New Issue