Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(installer): add --malloc_threshold arg

Browse Source 
Most systems do best with an explicit value for this the `MALLOC_MMAP_THRESHOLD_` env var, but some may want to change or unset this variable.

Valid values are a positive integer or "unset" to unset the variable.

Closes #6007.
This commit is contained in:
psychedelicious 2024-03-25 17:19:29 +11:00
parent 0f02a72cb9
commit 6d366fb519
1 changed files with 28 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
installer/templates/invoke.sh.in
View File

@ -46,8 +46,33 @@ if [ "$(uname -s)" == "Darwin" ]; then
export PYTORCH_ENABLE_MPS_FALLBACK=1
fi
# Avoid glibc memory fragmentation. See invokeai/backend/model_management/README.md for details.
export MALLOC_MMAP_THRESHOLD_=1048576
# Avoid glibc memory fragmentation. See #6007, #4784 and invokeai/backend/model_management/README.md for details.
# Some systems may need this to be set to a different value, so we may override this via command-line argument below.
export MALLOC_MMAP_THRESHOLD_=1048576 # 1MB
# This will be passed on to `invokeai-web`
PARAMS=()
# Parse command-line arguments
for arg in "$@"; do
if [[ $arg == --malloc_threshold=* ]]; then
# Re-set MALLOC_MMAP_THRESHOLD_ from the argument if provided
value="${arg#*=}"
if [[ $value == "unset" ]]; then
unset MALLOC_MMAP_THRESHOLD_
elif [[ $value =~ ^[0-9]+$ ]]; then
export MALLOC_MMAP_THRESHOLD_="$value"
else
echo "Invalid value for --malloc_threshold. Please provide a valid positive integer or 'unset'."
exit 1
fi
else
# Add other arguments to PARAMS
PARAMS+=("$arg")
fi
done
echo $MALLOC_MMAP_THRESHOLD_
# Primary function for the case statement to determine user input
do_choice() {
@ -55,7 +80,7 @@ do_choice() {
1)
clear
printf "Generate images with a browser-based interface\n"
invokeai-web $PARAMS
invokeai-web "${PARAMS[@]}"
;;
2)
clear