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 'development' into lstein-outcrop-improvements

Browse Source
This commit is contained in:
Lincoln Stein
2022-11-12 10:39:57 -05:00
committed by GitHub
parent bb084a844b 0a88243911
commit 9c218788e2
15 changed files with 371 additions and 241 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
ldm/generate.py
View File

@ -805,6 +805,10 @@ class Generate:
# the model cache does the loading and offloading
cache = self.model_cache
if not cache.valid_model(model_name):
print(f'** "{model_name}" is not a known model name. Please check your models.yaml file')
return self.model
cache.print_vram_usage()
# have to get rid of all references to model in order
@ -1032,7 +1036,9 @@ class Generate:
return True
return False
def _check_for_erasure(self, image):
def _check_for_erasure(self, image:Image.Image)->bool:
if image.mode not in ('RGBA','RGB'):
return False
width, height = image.size
pixdata = image.load()
colored = 0