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

add a ~/.invokeai file the first time we load

Browse Source 
- If there is not already a `.invokeai` file in the user's home directory
  the first time invoke.py runs, it will create an empty one with comments
  showing how to customize it.
This commit is contained in:
Lincoln Stein 2022-11-14 23:04:10 +00:00
parent 9fa3e28dd4
commit f62cc7db9d
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
ldm/invoke/args.py
View File

@ -173,7 +173,8 @@ class Args(object):
print(f'>> Initialization file {INITFILE} found. Loading...') print(f'>> Initialization file {INITFILE} found. Loading...')
sysargs.insert(0,f'@{INITFILE}') sysargs.insert(0,f'@{INITFILE}')
else: else:
print(f'>> Initialization file {INITFILE} not found. Applying default settings...') print(f'>> Initialization file {INITFILE} not found. Creating a new one...')
self._create_init_file(INITFILE)
self._arg_switches = self._arg_parser.parse_args(sysargs) self._arg_switches = self._arg_parser.parse_args(sysargs)
return self._arg_switches return self._arg_switches
except Exception as e: except Exception as e:
@ -365,6 +366,17 @@ class Args(object):
new_dict[k] = value2 if value2 is not None else value1 new_dict[k] = value2 if value2 is not None else value1
return new_dict return new_dict
def _create_init_file(self,initfile:str):
with open(initfile, mode='w', encoding='utf-8') as f:
f.write('''# InvokeAI initialization file
# Put frequently-used startup commands here, one or more per line
# Examples:
# --web --host=0.0.0.0
# --steps 20
# -Ak_euler_a -C10.0
'''
)
def _create_arg_parser(self): def _create_arg_parser(self):
''' '''
This defines all the arguments used on the command line when you launch This defines all the arguments used on the command line when you launch