From 9dd09d2fb73e237206459c2bb742ecb8e10d5a79 Mon Sep 17 00:00:00 2001
From: Ayaan Shah <ayaanns2@illinois.edu>
Date: Wed, 9 Aug 2023 13:12:58 -0700
Subject: [PATCH] added int typecasts

---
 zoedepth/models/base_models/midas.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/zoedepth/models/base_models/midas.py b/zoedepth/models/base_models/midas.py
index e26f85895..ec9478c73 100644
--- a/zoedepth/models/base_models/midas.py
+++ b/zoedepth/models/base_models/midas.py
@@ -170,6 +170,8 @@ def get_size(self, width, height):
 
     def __call__(self, x):
         width, height = self.get_size(*x.shape[-2:][::-1])
+        height = int(height)
+        width = int(width)
         return nn.functional.interpolate(x, (height, width), mode='bilinear', align_corners=True)
 
 class PrepForMidas(object):