danilw/krita-stablediffusion-webui-colab

Colab script for Krita Stable Diffusion plugin

Shadertoy page shadertoy.com/user/morimea itch.io page danilw.itch.io Vulkan shadertoy launcher vulkan-shadertoy-launcher Godot projects repository godot-utils-and-other y.svg

Hi there 👋

Graphic programming, GLSL and Godot projects.

📕 Blog Posts

• Float precision on GPU, bugs/features
• Launching 619 thousand Tetris on GPU
• Games in the GPU shaders
• Animations with shader programming
• github mirror of medium blog

Contact - message me anywhere, I'll respond.

Or s48gs.w <at> gmail {dot} com

krita-stablediffusion-webui-colab's Issues

[question] info

Cool, thanks for sharing this!

Do you plan to also support https://github.com/Interpause/auto-sd-paint-ext with Colab?

Do the new models also work without modifications? Just add the url? Has anyone tried them? (v1-5-pruned-emaonly.ckpt, sd-v1-5-inpainting.ckpt)

Regarding plugin-server encrypted communications, ngrok creates both http and https (same url) and https seems to work with the Krita plugin without modification. The patches mentioned in the readme for encryption are still necessary for privacy on the server (if the images are saved as a file on the server and if the prompts are saved in some log / cli history, or saved in some file associated with the images)

A change I made that maybe it can be useful to someone, to use the model from google drive:

## modified /cells/12/source:
@@ -1,8 +1,14 @@
 #@title Model
-user_header = f"\"Authorization: Bearer {huggingface_user_token}\""
-tflname = "/content/stable-diffusion-webui/"+local_file_name
+from google.colab import drive
+drive_path = "/content/drive"
+drive.mount(drive_path,force_remount=True)
+#/content/drive/MyDrive/AI/models/sd-v1-4.ckpt
+tflname = "/content/drive/MyDrive/AI/models/sd-v1-4.ckpt"
 if(not os.path.exists(tflname)):
-  !wget --header={user_header} $local_model_url
+  user_header = f"\"Authorization: Bearer {huggingface_user_token}\""
+  tflname = "/content/stable-diffusion-webui/"+local_file_name
+  if(not os.path.exists(tflname)):
+    !wget --header={user_header} $local_model_url
 if(os.path.exists("/content/stable-diffusion-webui/model.ckpt")):
   !rm "/content/stable-diffusion-webui/model.ckpt"
 !ln -s "$tflname" "/content/stable-diffusion-webui/model.ckpt"

It could probably be better added as an option.
In this way the download is not carried out every time the instance is restarted, the loading of the model, however, is a little slower when the service is started (and obviously takes up space in google drive)