Commit Graph
100 Commits
Author SHA1 Message Date
Manuel Schmid 00a27e7400 feat: add config and option for uov prompt type 2024-06-24 21:21:11 +02:00
Manuel Schmid 9f9fcfa7ad fix: start enhance from index 1, not 0 2024-06-24 20:45:50 +02:00
Manuel Schmid acb76adf67 feat: use last enhanced prompt for upscaling if provided
prevents replaced subjects from being replaced again with original subjects when upscaling due to using the original prompt
2024-06-24 20:45:05 +02:00
Manuel Schmid 1d0965f8c9 feat: rename enhance image input label, add link to documentation 2024-06-24 20:43:28 +02:00
Manuel Schmid fe1e4ce4e5 feat: bump version, add links to documentation 2024-06-23 18:53:42 +02:00
Manuel Schmid 25f4484c79 fix: correctly display current progress on non-enhance steps after skipping 2024-06-23 18:17:44 +02:00
Manuel Schmid 1ef6367635 feat: add option to process uov either before or after enhance 2024-06-23 17:41:21 +02:00
Manuel Schmid 7ecaee8285 fix: correctly track progress with image_number > 1 and not enhance 2024-06-23 17:40:20 +02:00
Manuel Schmid 54f25cb1ae feat: cleanup python package versions, remove diffusers
diffusers is not needed anymore as safety checker has been implemented natively
2024-06-23 16:22:08 +02:00
Manuel Schmid b094ac820b fix: only allow one image number if enhance image has been uploaded 2024-06-23 16:21:10 +02:00
Manuel Schmid bd3a29ba68 fix: use 0 steps when upscaling with skip_prompt_processing 2024-06-22 01:45:53 +02:00
Manuel Schmid 7ed98521ec fix: correctly call upscale vary 2024-06-22 01:06:40 +02:00
Manuel Schmid e8220bae49 feat: display intermediate results wqhen enhance is true 2024-06-22 01:06:22 +02:00
Manuel Schmid bef53fcf6e feat: automatically open tab enhance on enhance image upload 2024-06-22 00:30:47 +02:00
Manuel Schmid 18d294de63 feat: do not extend main prompt when enhance prompt is provided
reduces prompt consistency with input but greatly improves prompt accuracy
2024-06-22 00:14:32 +02:00
Manuel Schmid ec533f0c42 feat: rename arg --enable-describe-uov-image to --enable-auto-describe-image, add handling for enhance input image 2024-06-22 00:07:32 +02:00
Manuel Schmid ef236339f0 Merge branch 'main' into feature/multi-step-rendering 2024-06-22 00:01:40 +02:00
Manuel Schmid 722f170703 feat: update switch instructions 2024-06-21 23:58:18 +02:00
Manuel Schmid 6f7443276a feat: use casefold for comparing uov_method to flags.disabled
label "Disabled" is uppercase, convert to casefold before comparison
2024-06-21 23:52:11 +02:00
Manuel Schmid 40e1c82b74 feat: add enhance image input
use this so you don't have to modify an image before enhancement
2024-06-21 23:51:00 +02:00
Manuel Schmid c7a411a8c7 refactor: code cleanup 2024-06-21 22:43:03 +02:00
Manuel Schmid ebd5e24531 feat: allow upscaled input for enhance upscaling, correctly calculate steps 2024-06-21 22:42:43 +02:00
Manuel Schmid e0ed94aa6a Merge branch 'main' into feature/multi-step-rendering 2024-06-21 20:26:23 +02:00
Manuel Schmid 0b876eec20 Merge branch 'main_upstream' 2024-06-21 20:25:50 +02:00
Manuel Schmid 5bc6a0c650 fix: use correct parameter for upscale (fast) 2024-06-21 19:56:35 +02:00
Manuel Schmid 51dc53b5c8 feat: save and output upscaled image with method fast in enhance 2024-06-20 02:49:22 +02:00
Manuel Schmid eada51dee4 feat: optimize progress bar, now correctly uses uov steps and overrides 2024-06-20 02:43:14 +02:00
Manuel Schmid 5012fb7067 feat: display first enhance tab as #1 2024-06-20 02:20:49 +02:00
Manuel Schmid 4e575b9eb1 wip: add upscale or variation to enhance 2024-06-19 23:53:15 +02:00
Manuel Schmid 87b3cec7d4 fix: download correct inpaint engine in enhance 2024-06-19 21:31:01 +02:00
Manuel Schmid 6e3919f854 fix: adjust default visibility for detection prompt dataset 2024-06-19 01:01:46 +02:00
Manuel Schmid 0d817cc0d0 feat: add placeholder for detection prompt text input elements 2024-06-19 01:00:11 +02:00
Manuel Schmid 22e1b08e4f feat: add detection prompt dataset for quick selection 2024-06-19 00:58:55 +02:00
Manuel Schmid 3b55e64990 fix: remove unused path_adetailer from config 2024-06-19 00:45:50 +02:00
Manuel Schmid 53b27e4b74 release: bump version to 2..5.0-rc1 2024-06-19 00:21:57 +02:00
Manuel Schmid cef92ffffe Merge branch 'main_upstream' into feature/multi-step-rendering 2024-06-19 00:12:04 +02:00
Manuel Schmid 2c721e06ff feat: update performance hint for inpaint engine
inpaint engine model has only been trained for SDXL and no other performance LoRA, will produce bad results for inpaint larger areas
2024-06-19 00:02:29 +02:00
Manuel Schmid ff3dff8323 Revert "feat: add hint for performance when enhance inpaint engine is not none"
This reverts commit ed6c526986.
2024-06-18 23:52:13 +02:00
Manuel Schmid ed6c526986 feat: add hint for performance when enhance inpaint engine is not none 2024-06-18 23:51:36 +02:00
Manuel Schmid c0397bddbd feat: enable edit attention with ctrl + arrow keys for enhance prompts 2024-06-18 22:39:09 +02:00
Manuel Schmid 6cb0b2143b fix: correctly initialize inpaint_mask var, rename task styles index 2024-06-18 22:38:36 +02:00
Manuel Schmid dd866616bc refactor: code cleanup 2024-06-18 22:16:48 +02:00
Manuel Schmid c0f7c3a8ee feat: add erode or dilate and mask invert to enhance mask inpaint settings 2024-06-18 21:39:27 +02:00
Manuel Schmid 229ff81738 refactor: rename max_num_boxes to max_detections 2024-06-18 21:07:27 +02:00
Manuel Schmid b7fb42436c fix: correctly calculate current progress, display progressbar without any jumping 2024-06-18 12:35:41 +02:00
Manuel Schmid 1175df843b fix: correctly proceed progress bar within tasks
does not yet include jumping back for preparation steps
2024-06-18 11:48:42 +02:00
Manuel Schmid 3567c04918 wip: refactor code to make it more efficient
now first processes all tasks and then does enhancements
2024-06-17 21:46:00 +02:00
Manuel Schmid 24d66f6f77 Revert "feat: update GroundingDINO from SwinT (tiny) to SwinB (base)"
This reverts commit 7f3fd8f318.
2024-06-16 23:22:21 +02:00
Manuel Schmid 7f3fd8f318 feat: update GroundingDINO from SwinT (tiny) to SwinB (base) 2024-06-16 23:22:10 +02:00
Manuel Schmid 65cc25e5bf feat: add debug setting for enhance mask output 2024-06-16 22:54:44 +02:00
Manuel Schmid e300930298 feat: improve performance, do not reload the pipeline when prompts are the same as in the original task 2024-06-16 22:18:11 +02:00
Manuel Schmid 33a9c156ad fix: do not apply loras more than one time when not using lora in prompt 2024-06-16 21:40:39 +02:00
Manuel Schmid eeb1b79baa feat: optimize prompt translation 2024-06-16 21:40:07 +02:00
Manuel Schmid 9c93c18d0b fix: process other models than sam when using enhance 2024-06-16 21:05:10 +02:00
Manuel Schmid ff3418876d feat: refresh the whole pipeline, allows usage of inpaint and enhancement prompts 2024-06-16 20:44:35 +02:00
Manuel Schmid e1be3fa37a Merge branch 'main' into feature/multi-step-rendering 2024-06-16 18:54:52 +02:00
Manuel Schmid cecd6876a5 Merge branch 'hotfix/add-missing-method-in-performance' 2024-06-16 18:54:42 +02:00
Manuel Schmid 9ef7fe0843 hotfix: add missing method in performance enum 2024-06-16 18:52:20 +02:00
Manuel Schmid b64d6d393c fix: use != for literal comparison 2024-06-16 18:49:52 +02:00
Manuel Schmid 64fc025c50 Merge branch 'main' into feature/multi-step-rendering 2024-06-16 18:44:54 +02:00
Manuel Schmid af209cda53 fix: add workaround for same value in Steps IntEnum
(cherry picked from commit ea1562d078)
2024-06-16 18:41:17 +02:00
Manuel Schmid bf6820812f feat: only process enhance when checkbox is set
similar to input image
2024-06-16 18:39:40 +02:00
Manuel Schmid ea1562d078 fix: add workaround for same value in Steps IntEnum 2024-06-16 17:46:48 +02:00
Manuel Schmid e279d032ef Merge branch 'hotfix/correctly-filter-performance-lora' 2024-06-16 16:50:47 +02:00
Manuel Schmid 2704c90997 Merge branch 'hotfix/correctly-filter-performance-lora' into feature/multi-step-rendering 2024-06-16 16:50:26 +02:00
Manuel Schmid 6b17bf56d1 fix: correctly identify and remove performance LoRA
(cherry picked from commit b3e4506009)
(cherry picked from commit b947a5a4e6)
2024-06-16 16:50:04 +02:00
Manuel Schmid b947a5a4e6 fix: correctly identify and remove performance LoRA
(cherry picked from commit b3e4506009)
2024-06-16 16:48:35 +02:00
Manuel Schmid b3e4506009 fix: correctly identify and remove performance LoRA 2024-06-16 16:48:12 +02:00
Manuel Schmid 541fb2d445 feat: add support for enhance prompts 2024-06-16 15:58:27 +02:00
Manuel Schmid b585d9dfa7 fix: correctly count sam masks 2024-06-16 15:57:53 +02:00
Manuel Schmid ef9fd293ff wip: rename stage2 to enhance 2024-06-16 12:45:18 +02:00
Manuel Schmid a3dcd6e1fd wip: add inputs for positive and negative prompt for stage2 2024-06-16 12:39:05 +02:00
Manuel Schmid 71f5e70085 refactor: rename segmentation prompt to detection prompt
improved wording: prompt is only applied for GroundingDINO, not actually for segmentation
2024-06-16 12:30:49 +02:00
Manuel Schmid 8b21303542 wip: refactor UI for stage2 2024-06-15 01:38:22 +02:00
Manuel Schmid db0e0aaf59 wip: add inpaint options to stage2 2024-06-14 23:58:59 +02:00
Manuel Schmid a71739026b refactor: change wording of hint when nothing has been detected 2024-06-13 01:19:28 +02:00
Manuel Schmid f89d5a97b0 refactor: change wording of stage2 disclaimer, adjust gradio structure accordingly 2024-06-13 01:17:20 +02:00
Manuel Schmid f8f36828c7 feat: add disclaimer + skipping, stage2 won't properly work when used with inpaint or outpaint 2024-06-13 01:13:21 +02:00
Manuel Schmid dbc844804b feat: add handling for stage2_mask_sam_max_num_boxes and config 2024-06-12 22:16:02 +02:00
Manuel Schmid 9998b52dd2 feat: add gradio elements for input 2024-06-12 21:52:48 +02:00
Manuel Schmid 190c4b0a6f fix: do not set denoising_strength to async_worker dynamically
use parameter input and output
2024-06-12 21:51:49 +02:00
Manuel Schmid 2590230bcd Merge branch 'main' into feature/multi-step-rendering 2024-06-11 22:56:35 +02:00
Manuel Schmid 07de6dd5f8 Merge remote-tracking branch 'upstream/main' 2024-06-11 22:56:17 +02:00
Manuel Schmid e055510f1b fix: use correct mask dimensions 2024-06-11 00:12:02 +02:00
Manuel Schmid b5b33361c9 feat: only process sam_options.dino_erode_or_dilate when set 2024-06-10 23:47:09 +02:00
Manuel Schmid ab3f4df935 feat: use vit_b as default for experiment 2024-06-10 23:46:50 +02:00
Manuel Schmid ead1ed617a refactor: remove adetailer code 2024-06-10 23:46:25 +02:00
Manuel Schmid 29967d3a18 feat: add dino_erode_or_dilate and dino_debug again 2024-06-10 23:23:38 +02:00
Manuel Schmid 757863c023 feat: wrap sam model in model patcher for predict 2024-06-10 22:42:35 +02:00
Manuel Schmid 651f9c5cfd refactor: rename box_erode_or_dilate to dino_erode_or_dilate, add option dino_debug 2024-06-10 20:47:22 +02:00
Manuel Schmid b8578a080a fix: remove already declared config option 2024-06-10 20:46:25 +02:00
Manuel Schmid 3873892b0a feat: change default_inpaint_mask_sam_model to match sam model registry 2024-06-10 20:45:56 +02:00
Manuel Schmid ce1fb74270 feat: add download for sam models to config 2024-06-10 20:33:49 +02:00
Manuel Schmid 980563de9d feat: remove inpaint_mask_sam_quant 2024-06-10 20:32:12 +02:00
Manuel Schmid 8a81993940 wip: remove ultralytics, always use manual sam for image mask instead of rembg 2024-06-10 01:33:03 +02:00
Manuel Schmid 09e23f5509 refactor: add info to inpaint_mask_sam_prompt_text, rename variable for dino debugging 2024-06-09 23:09:46 +02:00
Manuel Schmid 57c049858c feat: add debug dino and mask dilate and erode 2024-06-09 22:31:41 +02:00
Manuel Schmid f2e7b65ed3 wip: add experiments mask generation, install segment_anything v1.0 2024-06-09 18:48:01 +02:00
Manuel Schmid 8087861743 refactor: reorder inpaint_mask_sam_model by model size ascending 2024-06-09 18:46:18 +02:00
Manuel Schmid ff9fa6c837 feat: add box_erode_or_dilate to generate_mask_from_image, expose more sam return values 2024-06-09 18:45:13 +02:00