Update app.py

app.py

@@ -69,4 +69,219 @@ with capture.capture_output() as cap:
   else:
     !pip install gradio==3.23 --no-deps -qq
 clear_output()
-inf('\u2714 Done','success', '50px')
+inf('\u2714 Done','success', '50px')
+
+#@markdown # Requirements
+
+print('Installing requirements...')
+
+with capture.capture_output() as cap:
+  %cd /content/ 
+  !wget -q -i https://raw.githubusercontent.com/TheLastBen/fast-stable-diffusion/main/Dependencies/A1111.txt
+  !dpkg -i *.deb
+  if not os.path.exists('/content/gdrive/'+mainpth+'/sd/stablediffusion'):
+    !tar -C /content/gdrive/$mainpth --zstd -xf sd_rep.tar.zst
+  !tar -C / --zstd -xf gcolabdeps.tar.zst
+  !rm *.deb | rm *.zst | rm *.txt
+  %env LD_PRELOAD=libtcmalloc.so
+  os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
+  os.environ['PYTHONWARNINGS'] = 'ignore'
+  if Use_Latest_Working_Commit:
+    !pip install gradio==3.16.2 --no-deps -qq
+  else:
+    !pip install gradio==3.23 --no-deps -qq
+
+clear_output()
+inf('\u2714 Done','success', '50px')
+
+#@markdown # Model Download/Load
+Model_Version = "1.5" #@param [ "1.5", "v1.5 Inpainting ", "V2.1-512px", "V2.1-768px"]
+
+Redownload_the_original_model = False #@param {type:"boolean"}
+
+def rmv():
+  !wget -q -O /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/models/Stable-diffusion/model.ckpt https://raw.githubusercontent.com/TheLastBen/fast-stable-diffusion/main/precompiled/README.md
+  !mv /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/models/Stable-diffusion/model.ckpt /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/models/Stable-diffusion/trashfile.f
+  time.sleep(2)
+  !rm /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/models/Stable-diffusion/trashfile.f
+  clear_output()
+
+if Redownload_the_original_model:
+  with capture.capture_output() as cap:
+    rmv()
+
+#@markdown Or
+Path_to_MODEL = "" #@param {type:"string"}
+#@markdown - Insert the full path of your trained model or to a folder containing multiple models
+
+#@markdown Or
+MODEL_LINK = "" #@param {type:"string"}
+safetensors = False #@param {type:"boolean"}
+Use_temp_storage = False #@param {type:"boolean"}
+
+
+
+def newmdl():
+
+    if not os.path.exists('/content/gdrive/'+mainpth+'/sd/stable-diffusion-'+blasphemy+'/models/Stable-diffusion/model.ckpt'):
+      %cd /content/
+      clear_output()
+      !git init
+      !git lfs install --system --skip-repo
+      !$fgitclone --branch fp16 "https://huggingface.co/runwayml/stable-diffusion-v1-5"
+      if os.path.exists('/content/stable-diffusion-v1-5/unet/diffusion_pytorch_model.bin'):
+        !wget -q -O stable-diffusion-v1-5/vae/diffusion_pytorch_model.bin https://huggingface.co/stabilityai/sd-vae-ft-mse/resolve/main/diffusion_pytorch_model.bin
+        !wget -q -O convertosd.py https://github.com/TheLastBen/fast-stable-diffusion/raw/main/Dreambooth/convertosd.py
+        !sed -i '201s@.*@    model_path = "/content/stable-diffusion-v1-5"@' /content/convertosd.py
+        !sed -i '202s@.*@    checkpoint_path= "/content/gdrive/{mainpth}/sd/stable-diffusion-{blasphemy}/models/Stable-diffusion/model.ckpt"@' /content/convertosd.py
+        clear_output()       
+        !python /content/convertosd.py
+        !rm /content/convertosd.py
+        if os.path.exists('/content/gdrive/'+mainpth+'/sd/stable-diffusion-'+blasphemy+'/models/Stable-diffusion/model.ckpt'):
+          model='/content/gdrive/'+mainpth+'/sd/stable-diffusion-'+blasphemy+'/models/Stable-diffusion/model.ckpt'
+          clear_output()
+          inf('\u2714 Done','success', '50px')
+        else:
+          inf('\u2718 Something went wrong, try again','danger', "250px")
+      else:
+        inf('\u2718 Something went wrong','danger', "200px")
+
+    else:
+      model='/content/gdrive/'+mainpth+'/sd/stable-diffusion-'+blasphemy+'/models/Stable-diffusion/model.ckpt'
+      clear_output()
+      inf('\u2714 Model already exists, check the box "Redownload_the_original_model" to redownload/download the V1.5','primary', '700px')
+
+    if os.path.exists('/content/.git'):
+      !rm -r /content/.git
+
+    if os.path.exists('/content/stable-diffusion-v1-5'):
+      !rm -r /content/stable-diffusion-v1-5
+
+    return model
+
+def V2():
+
+    if not os.path.exists('/content/gdrive/'+mainpth+'/sd/stable-diffusion-'+blasphemy+'/models/Stable-diffusion/model.ckpt'):
+      %cd /content/
+      clear_output()
+      !mkdir "/content/stable-diffusion-V2"
+      %cd "/content/stable-diffusion-V2"
+      !git init
+      !git lfs install --system --skip-repo
+      if Model_Version == "V2.1-768px":
+        !git remote add -f origin  "https://huggingface.co/stabilityai/stable-diffusion-2-1"
+      elif Model_Version == "V2.1-512px":
+        !git remote add -f origin  "https://huggingface.co/stabilityai/stable-diffusion-2-1-base"
+      !git config core.sparsecheckout true
+      !echo -e "scheduler\ntext_encoder\ntokenizer\nunet\nvae\nfeature_extractor\nmodel_index.json\n!*.safetensors" > .git/info/sparse-checkout
+      !git pull origin fp16
+      %cd /content
+      !wget -O convertosdv2.py https://github.com/TheLastBen/fast-stable-diffusion/raw/main/Dreambooth/convertosdv2.py
+      clear_output()
+      !python /content/convertosdv2.py --fp16 /content/stable-diffusion-V2 /content/gdrive/$mainpth/sd/stable-diffusion-$blasphemy/models/Stable-diffusion/model.ckpt
+      !rm /content/convertosdv2.py
+      if os.path.exists('/content/gdrive/'+mainpth+'/sd/stable-diffusion-'+blasphemy+'/models/Stable-diffusion/model.ckpt'):
+        model='/content/gdrive/'+mainpth+'/sd/stable-diffusion-'+blasphemy+'/models/Stable-diffusion/model.ckpt'
+        clear_output()
+        inf('\u2714 Done','success', '50px')
+      else:
+        inf('\u2718 Something went wrong, try again','danger', "250px")
+
+    else:
+      model='/content/gdrive/'+mainpth+'/sd/stable-diffusion-'+blasphemy+'/models/Stable-diffusion/model.ckpt'
+      clear_output()
+      inf('\u2714 Model already exists, check the box "Redownload_the_original_model" to redownload/download the V2','primary', '700px')
+
+    if os.path.exists('/content/.git'):
+      !rm -r /content/.git
+      !rm -r /content/convertosdv2.py
+    if os.path.exists('/content/stable-diffusion-V2'):
+      !rm -r /content/stable-diffusion-V2
+
+    return model
+
+def inpmdl():
+
+    if not os.path.exists('/content/gdrive/'+mainpth+'/sd/stable-diffusion-'+blasphemy+'/models/Stable-diffusion/sd-v1-5-inpainting.ckpt'):
+      %cd /content/
+      clear_output()
+      !git init
+      !git lfs install --system --skip-repo
+      !$fgitclone --branch fp16 "https://huggingface.co/runwayml/stable-diffusion-inpainting"
+      if os.path.exists('/content/stable-diffusion-inpainting'):
+        !$fgitclone "https://huggingface.co/stabilityai/sd-vae-ft-mse"
+        !rm -r /content/stable-diffusion-inpainting/vae
+        !mv /content/sd-vae-ft-mse /content/stable-diffusion-inpainting/vae        
+        !wget -O convertosd.py https://github.com/TheLastBen/fast-stable-diffusion/raw/main/Dreambooth/convertosd.py
+        !sed -i '201s@.*@    model_path = "/content/stable-diffusion-inpainting"@' /content/convertosd.py
+        !sed -i '202s@.*@    checkpoint_path= "/content/gdrive/{mainpth}/sd/stable-diffusion-{blasphemy}/models/Stable-diffusion/sd-v1-5-inpainting.ckpt"@' /content/convertosd.py
+        clear_output()       
+        !python /content/convertosd.py
+        !rm /content/convertosd.py
+        if os.path.exists('/content/gdrive/'+mainpth+'/sd/stable-diffusion-'+blasphemy+'/models/Stable-diffusion/sd-v1-5-inpainting.ckpt'):
+          model='/content/gdrive/'+mainpth+'/sd/stable-diffusion-'+blasphemy+'/models/Stable-diffusion/sd-v1-5-inpainting.ckpt'
+          clear_output()
+          inf('\u2714 Done','success', '50px')
+        else:
+          inf('\u2718 Something went wrong, try again','danger', "250px")
+      else:
+        inf('\u2718 Something went wrong','danger', "200px")
+
+
+    else:
+      model='/content/gdrive/'+mainpth+'/sd/stable-diffusion-'+blasphemy+'/models/Stable-diffusion/sd-v1-5-inpainting.ckpt'
+      clear_output()
+      inf('\u2714 inpainting model already exists','primary', '250px')
+
+    if os.path.exists('/content/.git'):
+      !rm -r /content/.git
+
+    if os.path.exists('/content/stable-diffusion-inpainting'):
+      !rm -r /content/stable-diffusion-inpainting
+
+    return model
+
+if (Path_to_MODEL !=''):
+  if os.path.exists(str(Path_to_MODEL)):
+    inf('\u2714 Using the trained model.','success', '200px')
+
+  else:
+      while not os.path.exists(str(Path_to_MODEL)):
+        inf('\u2718 Wrong path, use the colab file explorer to copy the path : ','danger', "400px")
+        Path_to_MODEL=input()
+      if os.path.exists(str(Path_to_MODEL)):
+        inf('\u2714 Using the trained model.','success', '200px')
+
+  model=Path_to_MODEL
+
+elif MODEL_LINK != "":
+  gdrv='/content/gdrive/'+mainpth+'/sd/stable-diffusion-'+blasphemy+'/models/Stable-diffusion'
+  tmp="/content"
+  pth=tmp if Use_temp_storage else gdrv
+  %cd $pth
+  clear_output()
+  if not safetensors:
+    modelname="model.ckpt"
+  else:
+    modelname="model.safetensors"
+  !gdown --fuzzy -O $modelname "$MODEL_LINK"
+  if os.path.exists(f'{pth}/{modelname}') and os.path.getsize(f'{pth}/{modelname}') > 1810671599:    
+      model=f'{pth}/{modelname}'
+      clear_output()
+      inf('\u2714 Model downloaded, using the trained model.','success', '350px')
+  else:
+    if Use_temp_storage:
+      !rm $pth/$modelname
+    else:
+      rmv()
+    inf('\u2718 Wrong link, check that the link is valid','danger', "300px")
+
+
+elif Model_Version=="1.5":
+  model=newmdl()
+
+elif Model_Version=="V2.1-512px" or Model_Version=="V2.1-768px":
+  model=V2()
+
+else:
+   model=inpmdl()