Don't download google drive

This commit is contained in:
Cyberes 2022-12-05 22:29:39 -07:00 committed by GitHub
parent 1ca5096639
commit c0b774f2d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 5 deletions

View File

@ -439,7 +439,7 @@
"source": [
"%store -r symlink_to_notebooks model_storage_dir repo_storage_dir\n",
"try:\n",
" test = [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]\n",
" test = [symlink_to_notebooks, SECOND SECOND 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]\n",
"except NameError as e:\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",
@ -940,9 +940,9 @@
"metadata": {},
"outputs": [],
"source": [
"%store -r symlink_to_notebooks model_storage_dir repo_storage_dir\n",
"%store -r model_storage_dir repo_storage_dir\n",
"try:\n",
" test = [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]\n",
" test = [model_storage_dir, repo_storage_dir]\n",
"except NameError as e:\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",
@ -979,7 +979,7 @@
"source": [
"%store -r model_storage_dir repo_storage_dir link_novelai_anime_vae search_paperspace_datasets\n",
"try:\n",
" test = [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]\n",
" test = [model_storage_dir, repo_storage_dir, link_novelai_anime_vae, search_paperspace_datasets]\n",
"except NameError as e:\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",
@ -1127,7 +1127,7 @@
"source": [
"%store -r model_storage_dir repo_storage_dir activate_xformers activate_deepdanbooru activate_medvram disable_pickle_check gradio_port gradio_auth ui_theme insecure_extension_access\n",
"try:\n",
" test = [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]\n",
" test = [model_storage_dir, repo_storage_dir, activate_xformers, activate_deepdanbooru, activate_medvram, disable_pickle_check, gradio_port, gradio_auth, ui_theme, insecure_extension_access]\n",
"except NameError as e:\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",
@ -1389,6 +1389,10 @@
"def pretty_exit(str):\n",
" print(str)\n",
" raise\n",
"\n",
"if 'drive.google.com' in model_uri:\n",
" pretty_exit(\"Can't parse Google Drive links. Download the file yourself then upload it to the machine.\")\n",
"\n",
"magnet_match = re.search(r'magnet:\\?xt=urn:btih:[A-Za-z0-9&=%.]*', model_uri)\n",
"web_match = re.search(r'(https?:\\/\\/(?:www\\.|(?!www))[a-zA-Z0-9][a-zA-Z0-9-]+[a-zA-Z0-9]\\.[^\\s]{2,}|www\\.[a-zA-Z0-9][a-zA-Z0-9-]+[a-zA-Z0-9]\\.[^\\s]{2,}|https?:\\/\\/(?:www\\.|(?!www))[a-zA-Z0-9]+\\.[^\\s]{2,}|www\\.[a-zA-Z0-9]+\\.[^\\s]{2,})', model_uri)\n",
"web_dl_file = None\n",