add free/paid tier options back in

This commit is contained in:
Drake Panzer 2023-01-26 21:35:13 -07:00
parent 72ca5a7523
commit 7345688b4c
No known key found for this signature in database
GPG Key ID: 194A1C358AACFC39
1 changed files with 15 additions and 7 deletions

View File

@ -69,10 +69,21 @@
"metadata": {},
"outputs": [],
"source": [
"model_storage_dir = '/storage/models' # Where to store your model checkpoints.\n",
"# Choose where to store your model checkpoints.\n",
"\n",
"# Free Tier\n",
"# model_storage_dir = '/tmp/stable-diffusion-models'\n",
"\n",
"# Paid Tier\n",
"# model_storage_dir = '/storage/models'\n",
"\n",
"\n",
"\n",
"# Optional path settings\n",
"repo_storage_dir = '/storage/stable-diffusion' # Where to store your Stable Diffusion-related files.\n",
"\n",
"export_storage_dir = '/notebooks/exports' # Where the generated images will be exported to.\n",
"\n",
"pip_cache_dir = None # The installer can cache pip wheels so you don't have to re-download them\n",
" # every time you start the machine. I recommed setting it to '/storage/pip/cache'\n",
"\n",
@ -111,8 +122,6 @@
" # If you don't set a password anyone can install and run arbitrary code on your machine!\n",
" # Instead, use gradio_auth which will automatically enable extensions when set.\n",
"\n",
"export_storage_dir = '/notebooks/exports' # Where the generated images will be exported to.\n",
"\n",
"gradio_queue = False # Uses gradio queue; experimental option; breaks restart UI button.\n",
"\n",
"# ===================================================================================================\n",
@ -1377,7 +1386,7 @@
" dl_web_file(model_uri)\n",
" # clean exit here\n",
" else:\n",
" print('Required HTTP headers are incorrect:', end='\\n\\n')\n",
" print('Required HTTP headers are incorrect. One of these needs to be correct:', end='\\n\\n')\n",
" print('Content-Type:', response.headers['content-type'].split(\";\")[0] if response.headers.get('content-type') else 'None')\n",
" print('Must be \"application/octet-stream\"', end='\\n\\n')\n",
" print('Content-Disposition:', response.headers['content-disposition'] if response.headers.get('content-disposition') else 'None')\n",
@ -1431,13 +1440,12 @@
"\n",
"import shutil\n",
"import requests\n",
"from pathlib import Path\n",
"!pip install moviepy==1.0.3\n",
"!apt update\n",
"!apt install -y potrace python3-tk\n",
"\n",
"def update_repo_if_not_exists(path, repo_clone_url, pre=None):\n",
" if pre is not None:\n",
" pre() \n",
"def update_repo_if_not_exists(path, repo_clone_url):\n",
" if not os.path.exists(path):\n",
" !git clone \"{repo_clone_url}\" \"{path}\"\n",
" else:\n",