Skip to content

Commit

Permalink
fix(nodes): MLSD needs inputs to be multiples of 64
Browse files Browse the repository at this point in the history
  • Loading branch information
psychedelicious committed Sep 11, 2024
1 parent 0af70fa commit 66489b7
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions invokeai/backend/image_util/mlsd/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,8 +48,8 @@ def run(self, image: Image.Image, score_threshold: float = 0.1, distance_thresho

height, width, _channels = np_img.shape

# This model requires the input image to have a resolution that is a multiple of 8
np_img = resize_to_multiple(np_img, 8)
# This model requires the input image to have a resolution that is a multiple of 64
np_img = resize_to_multiple(np_img, 64)
img_output = np.zeros_like(np_img)

with torch.no_grad():
Expand Down

0 comments on commit 66489b7

Please sign in to comment.