From fceb83c2b0b49ae845b5ded815b95e6eb69d9313 Mon Sep 17 00:00:00 2001
From: Abdullah Barhoum <ka70911@gmail.com>
Date: Fri, 16 Sep 2022 21:10:40 +0200
Subject: [PATCH] Dev hlky

---
 services/hlky/Dockerfile           |  4 +-
 services/hlky/webui_streamlit.yaml | 83 ++++++++++++++++++++++++------
 2 files changed, 70 insertions(+), 17 deletions(-)

diff --git a/services/hlky/Dockerfile b/services/hlky/Dockerfile
index 4f6fc47..7fd8865 100644
--- a/services/hlky/Dockerfile
+++ b/services/hlky/Dockerfile
@@ -29,7 +29,7 @@ RUN pip install -U --no-cache-dir pyperclip
 ARG BRANCH=master
 ARG SHA=833a91047df999302f699637768741cecee9c37b
 # ARG BRANCH=dev
-# ARG SHA=b4de6caf697d311c1238c15a4c863fa529a35522
+# ARG SHA=5f3d7facdea58fc4f89b8c584d22a4639615a2f8
 RUN <<EOF
 cd stable-diffusion
 git fetch
@@ -61,4 +61,4 @@ EXPOSE 7860
 # run, -u to not buffer stdout / stderr
 CMD /docker/mount.sh && \
   python3 -u scripts/webui.py --outdir /output --ckpt /cache/models/model.ckpt --ldsr-dir /latent-diffusion --inbrowser ${CLI_ARGS}
-  # STREAMLIT_SERVER_PORT=7860 python -m streamlit run scripts/webui_streamlit.py
+# STREAMLIT_SERVER_PORT=7860 python -m streamlit run scripts/webui_streamlit.py
diff --git a/services/hlky/webui_streamlit.yaml b/services/hlky/webui_streamlit.yaml
index d32e5fe..589f464 100644
--- a/services/hlky/webui_streamlit.yaml
+++ b/services/hlky/webui_streamlit.yaml
@@ -2,9 +2,12 @@
 general:
   gpu: 0
   outdir: /outputs
-  ckpt: "/cache/models/model.ckpt"
+
+  default_model: "Stable Diffusion v1.4"
+  default_model_config: "configs/stable-diffusion/v1-inference.yaml"
+  default_model_path: "/cache/models/model.ckpt"
   fp:
-    name: "embeddings/alex/embeddings_gs-11000.pt"
+    name:
   GFPGAN_dir: "./src/gfpgan"
   RealESRGAN_dir: "./src/realesrgan"
   RealESRGAN_model: "RealESRGAN_x4plus"
@@ -15,44 +18,90 @@ general:
   extra_models_cpu: False
   extra_models_gpu: False
   save_metadata: True
+  save_format: "png"
   skip_grid: False
   skip_save: False
   grid_format: "jpg:95"
-  save_format: "png"
   n_rows: -1
   no_verify_input: False
   no_half: False
+  use_float16: False
   precision: "autocast"
   optimized: False
-  optimized_turbo: False
+  optimized_turbo: True
+  optimized_config: "optimizedSD/v1-inference.yaml"
   update_preview: True
-  update_preview_frequency: 1
+  update_preview_frequency: 5
 
 txt2img:
   prompt:
   height: 512
   width: 512
-  cfg_scale: 5.0
+  cfg_scale: 7.5
   seed: ""
   batch_count: 1
   batch_size: 1
-  sampling_steps: 50
-  default_sampler: "k_lms"
+  sampling_steps: 30
+  default_sampler: "k_euler"
   separate_prompts: False
+  update_preview: True
+  update_preview_frequency: 5
   normalize_prompt_weights: True
   save_individual_images: True
   save_grid: True
   group_by_prompt: True
   save_as_jpg: False
-  use_GFPGAN: True
-  use_RealESRGAN: True
+  use_GFPGAN: False
+  use_RealESRGAN: False
   RealESRGAN_model: "RealESRGAN_x4plus"
   variant_amount: 0.0
   variant_seed: ""
+  write_info_files: True
+
+txt2vid:
+  default_model: "CompVis/stable-diffusion-v1-4"
+  custom_models_list:
+    [
+      "CompVis/stable-diffusion-v1-4",
+      "naclbit/trinart_stable_diffusion_v2",
+      "hakurei/waifu-diffusion",
+      "osanseviero/BigGAN-deep-128",
+    ]
+  prompt:
+  height: 512
+  width: 512
+  cfg_scale: 7.5
+  seed: ""
+  batch_count: 1
+  batch_size: 1
+  sampling_steps: 30
+  num_inference_steps: 200
+  default_sampler: "k_euler"
+  scheduler_name: "klms"
+  separate_prompts: False
+  update_preview: True
+  update_preview_frequency: 5
+  dynamic_preview_frequency: True
+  normalize_prompt_weights: True
+  save_individual_images: True
+  save_video: True
+  group_by_prompt: True
+  write_info_files: True
+  do_loop: False
+  save_as_jpg: False
+  use_GFPGAN: False
+  use_RealESRGAN: False
+  RealESRGAN_model: "RealESRGAN_x4plus"
+  variant_amount: 0.0
+  variant_seed: ""
+  beta_start: 0.00085
+  beta_end: 0.012
+  beta_scheduler_type: "linear"
+  max_frames: 1000
 
 img2img:
   prompt:
-  sampling_steps: 50
+  sampling_steps: 30
   # Adding an int to toggles enables the corresponding feature.
   # 0: Create prompt matrix (separate multiple prompts using |, and get all combinations of them)
   # 1: Normalize Prompt Weights (ensure sum of weights add up to 1.0)
@@ -65,11 +114,12 @@ img2img:
   # 8: jpg samples
   # 9: Fix faces using GFPGAN
   # 10: Upscale images using Real-ESRGAN
-  sampler_name: k_lms
+  sampler_name: "k_euler"
   denoising_strength: 0.45
   # 0: Keep masked area
   # 1: Regenerate only masked area
   mask_mode: 0
+  mask_restore: False
   # 0: Just resize
   # 1: Crop and resize
   # 2: Resize and fill
@@ -77,7 +127,7 @@ img2img:
   # Leave blank for random seed:
   seed: ""
   ddim_eta: 0.0
-  cfg_scale: 5.0
+  cfg_scale: 7.5
   batch_count: 1
   batch_size: 1
   height: 512
@@ -87,16 +137,19 @@ img2img:
   loopback: True
   random_seed_loopback: True
   separate_prompts: False
+  update_preview: True
+  update_preview_frequency: 5
   normalize_prompt_weights: True
   save_individual_images: True
   save_grid: True
   group_by_prompt: True
   save_as_jpg: False
-  use_GFPGAN: True
-  use_RealESRGAN: True
+  use_GFPGAN: False
+  use_RealESRGAN: False
   RealESRGAN_model: "RealESRGAN_x4plus"
   variant_amount: 0.0
   variant_seed: ""
+  write_info_files: True
 
 gfpgan:
   strength: 100