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

fix launcher shell script to use correct names for ti and merge functions

Browse Source
This commit is contained in:
Lincoln Stein 2023-02-03 09:45:57 -05:00
parent c79678a643
commit ca0f3ec0e4
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -47,11 +47,11 @@ if [ "$0" != "bash" ]; then
;; ;;
3) 3)
echo "Starting Textual Inversion:" echo "Starting Textual Inversion:"
exec textual_inversion --gui $@ exec invokeai-ti --gui $@
;; ;;
4) 4)
echo "Merging Models:" echo "Merging Models:"
exec merge_models --gui $@ exec invokeai-merge --gui $@
;; ;;
5) 5)
echo "Developer Console:" echo "Developer Console:"

4
ldm/invoke/merge_diffusers.py
View File

@ -7,6 +7,7 @@ Copyright (c) 2023 Lincoln Stein and the InvokeAI Development Team
import argparse import argparse
import os import os
import sys import sys
import traceback
from argparse import Namespace from argparse import Namespace
from pathlib import Path from pathlib import Path
from typing import List, Union from typing import List, Union
@ -372,7 +373,8 @@ def main():
run_cli(args) run_cli(args)
print(f">> Conversion successful. New model is named {args.merged_model_name}") print(f">> Conversion successful. New model is named {args.merged_model_name}")
except Exception as e: except Exception as e:
print(f"** An error occurred while merging the pipelines: {str(e)}") print(">> An error occurred:")
traceback.print_exc()
sys.exit(-1) sys.exit(-1)
except KeyboardInterrupt: except KeyboardInterrupt:
sys.exit(-1) sys.exit(-1)