Mirrors
/
stable-diffusion-paperspace
mirror of https://github.com/Engineer-of-Stuff/stable-diffusion-paperspace.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #52 from xoihazard/add-export-setting 

Add export directory setting
This commit is contained in:
Cyberes 2023-01-03 12:56:26 -07:00 committed by GitHub
parent be3d3cb8cc 3b825cf92b
commit bf943c9d9f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
StableDiffusionUI_Voldemort_paperspace.ipynb
View File

@ -71,11 +71,13 @@
"# symlink_to_notebooks = True # Enables the creation of symlinks back to /notebooks/\n", "# symlink_to_notebooks = True # Enables the creation of symlinks back to /notebooks/\n",
"# model_storage_dir = '/tmp/stable-diffusion/models' # Where the models will be downloaded to.\n", "# model_storage_dir = '/tmp/stable-diffusion/models' # Where the models will be downloaded to.\n",
"# repo_storage_dir = '/notebooks' # Where the repository will be downloaded to.\n", "# repo_storage_dir = '/notebooks' # Where the repository will be downloaded to.\n",
"# export_storage_dir = '/notebooks/exports' # Where the generated images will be exported to.\n",
"\n", "\n",
"# Paid Tier\n", "# Paid Tier\n",
"# symlink_to_notebooks = False\n", "# symlink_to_notebooks = False\n",
"# model_storage_dir = '/storage/models'\n", "# model_storage_dir = '/storage/models'\n",
"# repo_storage_dir = '/notebooks'\n", "# repo_storage_dir = '/notebooks'\n",
"# export_storage_dir = '/notebooks/exports'\n",
"\n", "\n",
"\n", "\n",
"# Other optional settings\n", "# Other optional settings\n",
@ -116,7 +118,7 @@
"\n", "\n",
"# ===================================================================================================\n", "# ===================================================================================================\n",
"# Save variables to Jupiter's temp storage so we can access it even if the kernel restarts.\n", "# Save variables to Jupiter's temp storage so we can access it even if the kernel restarts.\n",
"%store symlink_to_notebooks model_storage_dir repo_storage_dir activate_xformers link_novelai_anime_vae download_scripts activate_deepdanbooru activate_medvram disable_pickle_check gradio_port gradio_auth search_paperspace_datasets ui_theme insecure_extension_access" "%store symlink_to_notebooks model_storage_dir repo_storage_dir export_storage_dir activate_xformers link_novelai_anime_vae download_scripts activate_deepdanbooru activate_medvram disable_pickle_check gradio_port gradio_auth search_paperspace_datasets ui_theme insecure_extension_access"
] ]
}, },
{ {
@ -1192,24 +1194,29 @@
"outputs": [], "outputs": [],
"source": [ "source": [
"try:\n", "try:\n",
" %store -r model_storage_dir repo_storage_dir\n", " %store -r model_storage_dir repo_storage_dir export_storage_dir\n",
" test = [model_storage_dir, repo_storage_dir]\n", " test = [model_storage_dir, repo_storage_dir, export_storage_dir]\n",
"except NameError as e:\n", "except NameError as e:\n",
" print(\"There is an issue with your variables.\")\n", " print(\"There is an issue with your variables.\")\n",
" print(\"Please go back to the first block and make sure your settings are correct, then run the cell.\")\n", " print(\"Please go back to the first block and make sure your settings are correct, then run the cell.\")\n",
" print('Error:', e)\n", " print('Error:', e)\n",
" import sys\n", " import sys\n",
" sys.exit(1)\n", " sys.exit(1)\n",
"\n",
"import os\n",
"\n",
"if not os.path.exists(export_storage_dir):\n",
" os.makedirs(export_storage_dir)\n",
"\n",
"!if [ $(dpkg-query -W -f='${Status}' p7zip-full 2>/dev/null | grep -c \"ok installed\") = 0 ]; then sudo apt update && sudo apt install -y p7zip-full; fi # install 7z if it isn't already installed\n", "!if [ $(dpkg-query -W -f='${Status}' p7zip-full 2>/dev/null | grep -c \"ok installed\") = 0 ]; then sudo apt update && sudo apt install -y p7zip-full; fi # install 7z if it isn't already installed\n",
"from datetime import datetime\n", "from datetime import datetime\n",
"datetime_str = datetime.now().strftime('%m-%d-%Y_%H:%M:%S')\n", "datetime_str = datetime.now().strftime('%m-%d-%Y_%H:%M:%S')\n",
"%cd /notebooks/\n", "%cd \"{export_storage_dir}\"\n",
"!mkdir -p \"{datetime_str}/log\"\n", "!mkdir -p \"{datetime_str}/log\"\n",
"!cd \"{repo_storage_dir}/stable-diffusion-webui/log/\" && mv * \"/notebooks/{datetime_str}/log\"\n", "!cd \"{repo_storage_dir}/stable-diffusion-webui/log/\" && mv * \"{export_storage_dir}/{datetime_str}/log\"\n",
"!cd \"{repo_storage_dir}/stable-diffusion-webui/outputs/\" && mv * \"/notebooks/{datetime_str}\"\n", "!cd \"{repo_storage_dir}/stable-diffusion-webui/outputs/\" && mv * \"{export_storage_dir}/{datetime_str}\"\n",
"!TEMP=\"/notebooks/{datetime_str}\" # find command has issues with ipynb variables??\n", "!find \"{export_storage_dir}/{datetime_str}\" -name .ipynb_checkpoints -exec rm -rf {{}} +\n",
"!find \"{TEMP}\" -name .ipynb_checkpoints -exec rm -rf \"{}\" +\n", "!7z a -t7z -m0=lzma2 -mx=9 -mfb=64 -md=32m -ms=on \"{datetime_str}.7z\" \"{export_storage_dir}/{datetime_str}/\""
"!7z a -t7z -m0=lzma2 -mx=9 -mfb=64 -md=32m -ms=on \"{datetime_str}.7z\" \"/notebooks/{datetime_str}/\""
] ]
}, },
{ {
@ -1229,8 +1236,8 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"# !rm -rf \"/notebooks/{datetime_str}/\"\n", "# !rm -rf \"{export_storage_dir}/{datetime_str}/\"\n",
"# !echo Deleted /notebooks/{datetime_str}/" "# !echo Deleted {export_storage_dir}/{datetime_str}/"
] ]
}, },
{ {