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

fix: assert the return depth map to be a PIL image

Browse Source
This commit is contained in:
blessedcoolant 2024-07-31 23:22:01 +05:30
parent b4cf78a95d
commit 95dde802ea
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/image_util/depth_anything/depth_anything_pipeline.py
View File

@ -17,7 +17,7 @@ class DepthAnythingPipeline(RawModel):
def generate_depth(self, image: Image.Image, resolution: int = 512):
image_width, image_height = image.size
depth_map = self.pipeline(image)["depth"]
depth_map = cast(Image.Image, depth_map)
assert isinstance(depth_map, Image.Image)
new_height = int(image_height * (resolution / image_width))
depth_map = depth_map.resize((resolution, new_height))