Commit Graph

95 Commits (648025ba6bc99df175d896a6aaf793cf72574b1c)

Author SHA1 Message Date
Chenlei Hu 648025ba6b
Depth Anything V2 Support (#2977)
* Add depth_anything_v2.py

* Update preprocessor_compiled.py

Added depth_anything_v2 preprocessor

* Update processor.py

Added depth_anything_v2 preprocessor

* Update depth_test.py

Added depth_anything_v2 to general_depth_modules

* Update modules_test.py

Added depth_anything_v2 to expected_module_names & expected_module_alias

* Update install.py

Added depth_anything_v2 to install.py

* Update README.md

* Fix model URL

---------

Co-authored-by: Mackinations.Ai <133395980+MackinationsAi@users.noreply.github.com>
2024-07-03 13:28:40 -04:00
Chenlei Hu 59c43499ea
Revert "Set default value of ControlNetUnit.enabled to True (#2912)" (#2919)
This reverts commit a5457dcc1c.
2024-05-20 12:32:08 -04:00
Chenlei Hu a5457dcc1c
Set default value of ControlNetUnit.enabled to True (#2912) 2024-05-19 21:22:14 -04:00
Chenlei Hu 8a8a42e416
Fix test expectation (#2906) 2024-05-19 14:07:41 -04:00
Chenlei Hu 58f620c921
Add PuLID VRAM leak reproduction tests (#2891)
* Add PuLID detect test

* Add PuLID generation test
2024-05-14 22:15:27 -04:00
Chenlei Hu e33c046158
Validate ControlNetUnit using pydantic (#2847)
* Add Pydantic ControlNetUnit

Add test config

Add images field

Adjust image field handling

fix various ui

Fix most UI issues

accept greyscale image/mask

Fix infotext

Fix preset

Fix infotext

nit

Move infotext parsing test

Remove preset

Remove unused js code

Adjust test payload

By default disable unit

refresh enum usage

Align resize mode

change test func name

remove unused import

nit

Change default handling

Skip bound check when not enabled

Fix batch

Various batch fix

Disable batch hijack test

adjust test

fix test expectations

Fix unit copy

nit

Fix test failures

* Change script args back to ControlNetUnit for compatibility

* import enum for compatibility

* Fix unit test

* simplify unfold

* Add test coverage

* handle directly set np image

* re-enable batch test

* Add back canvas scribble support

* nit

* Fix batch hijack test
2024-05-06 15:40:34 -04:00
Chenlei Hu 1b95e476ec
Adjust test template's threshold a/b value (#2852) 2024-05-05 19:15:39 -04:00
Chenlei Hu 45a49daa26
fix test (#2851) 2024-05-05 16:38:47 -04:00
Chenlei Hu 7b4571e20b
Add effective_region_mask for ControlNet [API] (#2818)
* Add control model type check

* Implement region mask for ControlNet
2024-04-27 17:05:28 -04:00
Chenlei Hu cb8fd242b1
Add effective_region_mask for ipadapter [API] (#2785)
* Add effective_region_mask for ipadapter [API]

* Fix various issues & Add tests

* nit
2024-04-26 23:24:23 -04:00
Chenlei Hu 22f63dc2dc
Support lama inpaint and CLIP mask in `controlnet/detect` API endpoint (#2809)
* api

* notes

* Add tests

* nit

* nit

* nit

* Fix shuffle
2024-04-25 08:50:03 -04:00
Chenlei Hu 07c2417ec0
Fix expected status code (#2808) 2024-04-24 22:28:13 -04:00
Chenlei Hu f7707d8ff5
Fix detect test for non-image return values (#2807)
* Handle tensor output from detect/ endpoint

* nit
2024-04-24 22:04:05 -04:00
Chenlei Hu f988edcb95
Add control types test (#2805) 2024-04-24 17:20:57 -04:00
Chenlei Hu ed373ed0e1
Add IPAdapter advanced weighting [Instant Style] (#2767)
* rename variable

* nit

* add transformer index

* nit

* API support

* Fix transformer_index alignment issue

* Add weight control panel

* fix sdxl

* free edit

* clear advanced weighting

* nit

* Add assertion
2024-04-20 01:14:22 -04:00
Chenlei Hu 282c619696
Fix inpaint preprocessors (#2763)
* wip

* adjust test
2024-04-18 10:30:48 -04:00
Chenlei Hu 442398bb9f
Port SDForge's preprocessor structure (#2754)
* wip

* wip2

* wip3

* fix issues

* Add scribble xdog

* port legacy processors

* nit

* Add tests

* Fix modules test

* Add back normal_dsine

* Remove legacy code

* Remove code

* Add tests

* rename param

* Linter ignore

* fix is_image

* fix is_image

* nit

* nit

* Better assertion message

* Add back ip-adapter-auto

* Add test

* Fix various tag matching

* fix

* Add back preprocessor cache

* Add back sparse ctrl

* fix test failure

* Add log
2024-04-17 22:28:56 -04:00
Chenlei Hu bbcae309d1
Save/Use CLIP output in API (#2590)
* Add API support

* Add assertion

* Add test

* nit
2024-04-15 15:08:51 -04:00
Chenlei Hu 8af7577c45
Change API processor_res/threshold_a/threshold_b default values (#2746)
* Change API threshold_a/threshold_b/res default values

* move code

* Add tests

* update test

* Add log_context for testing

* More test

* nit

* encoding

* Change cq log encoding
2024-04-15 09:46:14 -04:00
Chenlei Hu a5f9d59717
Add clip mask support (#2721)
* Add clip mask support

* nit

* Add tests
2024-03-31 12:13:51 -04:00
Chenlei Hu f6f36e3bbf
Add 'image_mask' field for compatibility with SD Forge (#2717)
* Add 'image_mask' field for compatibility with SD Forge

* Merge img2img test and txt2img tests

* Reduce target size to speed up CQ
2024-03-30 17:32:06 -04:00
Chenlei Hu f7aa688402
Fix seg preprocessor (#2716) 2024-03-30 15:36:41 -04:00
Chenlei Hu 4e9ac19cbc
Migrate detect_test and improve coverage (#2715)
* Migrate detection tests

* Add test results

* Skip cq disabled tests
2024-03-30 15:33:57 -04:00
Chenlei Hu a773a1a816
Real multi-inputs for all IP-Adapters (#2594)
* Add adapter tests

* real multi inputs

* multi-inputs

* Fix hook

* Use real multi for instant id
2024-01-28 12:44:40 -05:00
Chenlei Hu 14ab5e4ff6
Revert " Real multi-inputs unit" (#2583) 2024-01-25 08:52:10 -05:00
Chenlei Hu 6a1d882e89
Real multi-inputs unit (#2578)
* Add adapter tests

* real multi inputs
2024-01-24 20:19:11 -05:00
Chenlei Hu 482288e5bf
Update ipadapter tests to test negative prompt (#2570) 2024-01-23 19:14:20 -05:00
Chenlei Hu e5adb1697d
Add depth_anything (#2544)
*  Add depth_anything

* Fix various issues

* Add tests
2024-01-22 16:14:26 -05:00
Chenlei Hu 25e41dc63b
Real multi-inputs ControlNet unit (#2539)
* real multi-inputs

* add tests

* nit

* 🐛 Fix negative prompt issue
2024-01-22 02:27:37 -05:00
Chenlei Hu a79661e0db
🚚 Move inpaint crop logic as a function (#2534)
* Add tests

* 🔨 Extract crop logic
2024-01-21 19:00:28 -05:00
Chenlei Hu 5fde5430de
Support IP-Adapter faceid portrait (#2529)
* Add ip adapter tests

* Refactor faceid

* support portrait

*  more testing
2024-01-21 12:03:43 -05:00
Chenlei Hu 8870e35682
🐛 Fix outpaint 2 (#2485)
* 🐛 Fix outpaint

* 🐛 Fix lama outpaint
2024-01-16 12:36:04 -05:00
Chenlei Hu af9c3cdb9c
🔨 Refactor choose_input_image logic (#2468)
*  Add inpaint full cov tests

* More tests

* 🔨 Refactor choose input image logic

*  Update unittest expectation

* nits
2024-01-14 18:58:39 -05:00
Chenlei Hu 8bd8057cca
🐛 Fix hr_option parsing (#2398) 2024-01-03 08:50:57 -05:00
Chenlei Hu 7aaf263208
Add hr_option (Both, High res only, Low res only) (#2379)
*  Add hr_option

🎨 Only show when relevant

 Add unit test

* 🐛 fix per-unit
2024-01-01 00:06:55 -05:00
Chenlei Hu 11019f4820
Support advanced_weighting in API (#2368)
*  Support advanced_weighting in API

* 📝 add note

* 🐛 nits

* 📝 Add examples
2023-12-30 17:35:52 -05:00
liubo0902 b7a9bea231
fix image invert detect (#2356)
* fix image invert detect

*  Add unittest

---------

Co-authored-by: huchenlei <chenlei.hu@mail.utoronto.ca>
2023-12-24 21:31:51 -05:00
青龍聖者@bdsqlsz 79948d8c29
Add Animal Pose Preprocessor (RTMPose AP-10K) (#2293)
* add support for animal pose

* Update __init__.py

* fix typo

* flatten json result list

* format files

* remove unused functions

* 🚧 Support edit

* nits

* fix test expecetations

---------

Co-authored-by: huchenlei <chenlei.hu@mail.utoronto.ca>
2023-12-23 15:55:21 -05:00
Chenlei Hu 0b713ca5a5
🔧 Detect ControlNet lora for SD15 (#2336) 2023-12-15 23:57:49 -05:00
Chenlei Hu 27c78e3fec
🐛 Set default model/module to string values (#2335) 2023-12-15 21:54:16 -05:00
Chenlei Hu a13bd2febe
Add render_openposes_json API endpoint (#2325)
* Add render_openpose_json API endpoint

* nits
2023-12-12 22:07:52 -05:00
Chenlei Hu 54dfb899d5
Ignore unrecognized params (#2313) 2023-12-09 23:17:03 -05:00
Chenlei Hu 0efcf8c993
🐛 Handle None controlnet model in version check (#2272) 2023-11-19 23:51:39 -05:00
Chenlei Hu aa26871b42
More tests on SDXL models (#2264) 2023-11-18 17:48:56 -05:00
青龍聖者@bdsqlsz 711203e88e
Add support for IP Adapter Full Face (#2252)
* Add support for IP Adapter Full Face

*  Add unittests

* nits

---------

Co-authored-by: huchenlei <chenlei.hu@mail.utoronto.ca>
2023-11-12 21:53:17 -05:00
Chenlei Hu f14653a10e
Verify nunber of response images (#2255) 2023-11-12 19:56:50 -05:00
Chenlei Hu 9e61022428
🔧 Support SDXL in tests (#2254)
* 🔧 Support SDXL in tests

* 📝 docs

* nits
2023-11-12 14:12:54 -05:00
Chenlei Hu 457da74dc7
Add new API option send_detected_map (#2251) 2023-11-11 19:06:06 -05:00
Chenlei Hu 27c522029e
Fix CI tests (#2220)
* 🔧 setup CI

* 🐛 Fix path

* cd

* set working dir

* fix openpose tests

* Fix web_api tests

* Fix batch hijack tests

* change coverage dir

* 📝 Add doc
2023-11-04 00:48:22 -04:00
Abdullah Alfaraj b4cdd10283
[API]: expose the filter functionality to api (#1667)
* [API]: expose the filter functionality to the api

* add basic unit_test for the filter api endpoint

* rename filter_list to filter

* 🚧 Return all filters throught API call

---------

Co-authored-by: huchenlei <chenlei.hu@mail.utoronto.ca>
2023-07-17 23:38:50 -04:00