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

(minor) Add None check to fix static type checking error.

Browse Source
This commit is contained in:
Ryan Dick 2024-08-01 09:51:53 -04:00
parent f5cfdcf32d
commit 63581ec980
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
invokeai/backend/image_util/grounding_dino/grounding_dino_pipeline.py
View File

@ -18,6 +18,7 @@ class GroundingDinoPipeline(RawModel):
def detect(self, image: Image.Image, candidate_labels: list[str], threshold: float = 0.1) -> list[DetectionResult]:
results = self._pipeline(image=image, candidate_labels=candidate_labels, threshold=threshold)
assert results is not None
results = [DetectionResult.model_validate(result) for result in results]
return results