InvokeAI/scripts
psychedelicious 4602efd598
feat: add profiler util (#5601)
* feat(config): add profiling config settings

- `profile_graphs` enables graph profiling with cProfile
- `profiles_dir` sets the output for profiles

* feat(nodes): add Profiler util

Simple wrapper around cProfile.

* feat(nodes): use Profiler in invocation processor

* scripts: add generate_profile_graphs.sh script

Helper to generate graphs for profiles.

* pkg: add snakeviz and gprof2dot to dev deps

These are useful for profiling.

* tests: add tests for profiler util

* fix(profiler): handle previous profile not stopped cleanly

* feat(profiler): add profile_prefix config setting

The prefix is used when writing profile output files. Useful to organise profiles into sessions.

* tidy(profiler): add `_` to private API

* feat(profiler): simplify API

* feat(profiler): use child logger for profiler logs

* chore(profiler): update docstrings

* feat(profiler): stop() returns output path

* chore(profiler): fix docstring

* tests(profiler): update tests

* chore: ruff
2024-01-31 10:51:57 +00:00
..
configure_invokeai.py chore: ruff format 2023-11-11 10:55:40 +11:00
generate_profile_graphs.sh feat: add profiler util (#5601) 2024-01-31 10:51:57 +00:00
images2prompt.py isort wip 2 2023-09-12 13:01:58 -04:00
invokeai-cli.py isort wip 2 2023-09-12 13:01:58 -04:00
invokeai-model-install.py remove additional unused scripts 2023-09-02 10:05:29 -04:00
invokeai-web.py isort wip 2 2023-09-12 13:01:58 -04:00
make_models_markdown_table.py isort wip 2 2023-09-12 13:01:58 -04:00
probe-model.py fix import unsorted import block issues in the tests 2023-11-26 13:37:47 -05:00
pypi_helper.py Apply black 2023-07-27 10:54:01 -04:00
scan_models_directory.py chore: ruff check - fix flake8-comprensions 2023-11-11 10:55:23 +11:00
sd-metadata.py isort wip 2 2023-09-12 13:01:58 -04:00