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 attention map images to output object

Browse Source
This commit is contained in:
Lincoln Stein 2023-03-11 08:05:56 -05:00
parent 7e76eea059
commit 675dd12b6c
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/generator/base.py
View File

@ -64,6 +64,7 @@ class InvokeAIGeneratorOutput:
image: Image image: Image
seed: int seed: int
model_hash: str model_hash: str
attention_maps_images: List[Image]
params: Namespace params: Namespace
# we are interposing a wrapper around the original Generator classes so that # we are interposing a wrapper around the original Generator classes so that
@ -147,6 +148,7 @@ class InvokeAIGenerator(metaclass=ABCMeta):
output = InvokeAIGeneratorOutput( output = InvokeAIGeneratorOutput(
image=results[0][0], image=results[0][0],
seed=results[0][1], seed=results[0][1],
attention_maps_images=results[0][2],
model_hash = model_hash, model_hash = model_hash,
params=Namespace(**generator_args), params=Namespace(**generator_args),
) )
@ -366,7 +368,7 @@ class Generator:
if self.safety_checker is not None: if self.safety_checker is not None:
image = self.safety_checker.check(image) image = self.safety_checker.check(image)
results.append([image, seed]) results.append([image, seed, attention_maps_images])
if image_callback is not None: if image_callback is not None:
attention_maps_image = ( attention_maps_image = (