Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add STARTUP_CMD env var and IPEXRUN support to gui.sh #1662

Merged
merged 3 commits into from
Nov 11, 2023

Conversation

Disty0
Copy link
Contributor

@Disty0 Disty0 commented Nov 6, 2023

Add STARTUP_CMD and STARTUP_CMD_ARGS env vars so we can use custom scripts to run kohya_gui.py instead of Python.

Example use of STARTUP_CMD and STARTUP_CMD_ARGS env vars with ipexrun:
STARTUP_CMD=ipexrun STARTUP_CMD_ARGS="--multi-task-manager taskset --memory-allocator jemalloc" ./gui.sh

Add ipexrun support for Intel ARC GPUs.
ipexrun fixes some severe system memory leak issues and fixes constant nans with Full BF16 Training on Intel ARC.
It can cause some issues with old distros so setting DISABLE_IPEXRUN env var will disable ipexrun.
Setting STARTUP_CMD will disable ipexrun.
Setting STARTUP_CMD_ARGS will replace ipexrun args if ipexrun is enabled.

@Disty0 Disty0 changed the title Add STARTUP_CMD and STARTUP_CMD_ARGS env var to gui.sh Add STARTUP_CMD env var and IPEXRUN support to gui.sh Nov 6, 2023
@bmaltais
Copy link
Owner

Thank you for the submision. Look good. I will merge to dev2 and release this weekend.

@bmaltais bmaltais merged commit 609e184 into bmaltais:dev2 Nov 11, 2023
1 check failed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants