File size: 1,772 Bytes
7112279 74c5e9a 7112279 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | {
"cells": [
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"import threading\n",
"import yaml\n",
"\n",
"with open(\"/content/IAekovnI/db/configs/models.yaml\", \"r\") as f:\n",
" models = yaml.safe_load(f)\n",
"\n",
"new_models = {\n",
" \"sd-1/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting\": {\n",
" \"path\": \"/content/IAekovnI/db/models/sd-1/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting.safetensors\",\n",
" \"config\": \"/content/IAekovnI/db/configs/stable-diffusion/v1-inpainting-inference.yaml\",\n",
" \"variant\": \"inpaint\",\n",
" \"format\": \"checkpoint\",\n",
" },\n",
"}\n",
"\n",
"if \"sd-1/main/chilloutmix_NiPrunedFp32Fix-inpainting.inpainting\" not in models:\n",
" models.update(new_models)\n",
"\n",
"with open(\"/content/IAekovnI/db/configs/models.yaml\", \"w\") as f:\n",
" yaml.safe_dump(models, f)\n",
"\n",
"def start_tunnle():\n",
" if use_ngrok:\n",
" from pyngrok import ngrok\n",
" ngrok_tunnel = ngrok.connect(9090, \"http\")\n",
" print(\"ngrok_tunnel:\", ngrok_tunnel)\n",
" if use_cloudflare:\n",
" from pycloudflared import try_cloudflare\n",
" cloudflare_url = try_cloudflare(9090, verbose=False)\n",
" print(\"cloudflare_tunnel:\", cloudflare_url)\n",
"\n",
"threading.Thread(target=start_tunnle, daemon=True).start()\n",
"\n",
"!python /content/IAekovnI/scripts/invokeai-web.py --root /content/IAekovnI/db"
]
}
],
"metadata": {
"language_info": {
"name": "python"
}
},
"nbformat": 4,
"nbformat_minor": 2
}
|