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

Merge branch 'main' into pr/2561

Browse Source
This commit is contained in:
blessedcoolant 2023-02-13 23:48:18 +13:00
commit 9cbf9d52b4
2 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ldm/invoke/merge_diffusers.py
View File

@ -463,7 +463,7 @@ def main():
"** You need to have at least two diffusers models defined in models.yaml in order to merge" "** You need to have at least two diffusers models defined in models.yaml in order to merge"
) )
else: else:
print(f"** A layout error has occurred: {str(e)}") print(f"** Not enough room for the user interface. Try making this window larger.")
sys.exit(-1) sys.exit(-1)
except Exception as e: except Exception as e:
print(">> An error occurred:") print(">> An error occurred:")

20
ldm/invoke/training/textual_inversion.py
View File

@ -439,19 +439,23 @@ def main():
do_front_end(args) do_front_end(args)
else: else:
do_textual_inversion_training(**vars(args)) do_textual_inversion_training(**vars(args))
except widget.NotEnoughSpaceForWidget as e:
if str(e).startswith("Height of 1 allocated"):
print(
"** You need to have at least one diffusers models defined in models.yaml in order to train"
)
else:
print(f"** A layout error has occurred: {str(e)}")
sys.exit(-1)
except AssertionError as e: except AssertionError as e:
print(str(e)) print(str(e))
sys.exit(-1) sys.exit(-1)
except KeyboardInterrupt: except KeyboardInterrupt:
pass pass
except (widget.NotEnoughSpaceForWidget, Exception) as e:
if str(e).startswith("Height of 1 allocated"):
print(
"** You need to have at least one diffusers models defined in models.yaml in order to train"
)
elif str(e).startswith('addwstr'):
print(
'** Not enough window space for the interface. Please make your window larger and try again.'
)
else:
print(f"** A layout error has occurred: {str(e)}")
sys.exit(-1)
if __name__ == "__main__": if __name__ == "__main__":