Skip to content

Commit

Permalink
small fix
Browse files Browse the repository at this point in the history
  • Loading branch information
ddPn08 committed Dec 19, 2022
1 parent 21e4a81 commit e3764d4
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions automatic1111.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,8 @@
"source": [
"! nvidia-smi\n",
"! nvcc -V\n",
"! free -h"
"! free -h\n",
"! lsb_release -a"
]
},
{
Expand Down Expand Up @@ -134,6 +135,7 @@
" print(\"...reverting to default path variables\")\n",
"\n",
"if os.path.exists(f\"{data_dir}/script.pre.sh\"):\n",
" ! chmod +x {data_dir}/script.pre.sh\n",
" ! {data_dir}/script.pre.sh\n",
"\n",
"models_path = f\"{data_dir}/models\"\n",
Expand All @@ -158,7 +160,7 @@
" if mount_google_drive and not mount_success:\n",
" print(f\"Downloading model to {models_path} due to gdrive mount error\")\n",
" elif not model_filename or not os.path.exists(f\"{data_dir}/models/{model_filename}\"):\n",
" ! curl -LJ {model_url} -o {models_path}/{model_filename} {'-H \"Authorization: Bearer ' + auth_token + '\"' if auth_token else \"\"}\n",
" ! curl -LJ {model_url} -o {models_path}/Stable-diffusion/{model_filename} {'-H \"Authorization: Bearer ' + auth_token + '\"' if auth_token else \"\"}\n",
" else:\n",
" print(\"Model already downloaded, moving to next step\")"
]
Expand Down Expand Up @@ -343,6 +345,7 @@
"os.environ[\"LD_LIBRARY_PATH\"] = f\"{os.environ['LD_LIBRARY_PATH']}:/usr/local/envs/automatic/lib\"\n",
"\n",
"if os.path.exists(f\"{data_dir}/script.post.sh\"):\n",
" ! chmod +x {data_dir}/script.post.sh\n",
" ! {data_dir}/script.post.sh\n",
"\n",
"if store_env_gdrive and mount_google_drive and mount_success:\n",
Expand Down

0 comments on commit e3764d4

Please sign in to comment.