restore access token-saving code

This commit is contained in:
Lincoln Stein 2023-07-18 22:58:56 -04:00
parent 0f18898865
commit 0f7e329e76

View File

@ -560,7 +560,6 @@ def edit_opts(program_opts: Namespace, invokeai_opts: Namespace) -> argparse.Nam
editApp.run() editApp.run()
return editApp.new_opts() return editApp.new_opts()
def default_startup_options(init_file: Path) -> Namespace: def default_startup_options(init_file: Path) -> Namespace:
opts = InvokeAIAppConfig.get_config() opts = InvokeAIAppConfig.get_config()
if not init_file.exists(): if not init_file.exists():
@ -664,6 +663,9 @@ def write_opts(opts: Namespace, init_file: Path):
with open(init_file,'w', encoding='utf-8') as file: with open(init_file,'w', encoding='utf-8') as file:
file.write(new_config.to_yaml()) file.write(new_config.to_yaml())
if opts.hf_token:
HfLogin(opts.hf_token)
# ------------------------------------- # -------------------------------------
def default_output_dir() -> Path: def default_output_dir() -> Path:
return config.root_path / "outputs" return config.root_path / "outputs"