diff --git a/models/vista3d/scripts/inferer.py b/models/vista3d/scripts/inferer.py index 7232fe9e..6c230916 100644 --- a/models/vista3d/scripts/inferer.py +++ b/models/vista3d/scripts/inferer.py @@ -101,7 +101,7 @@ def __call__( ) else: val_outputs = SlidingWindowInfererAdapt( - roi_size=self.roi_size, sw_batch_size=self.sw_batch_size, with_coord=True, padding_mode='replicate' + roi_size=self.roi_size, sw_batch_size=self.sw_batch_size, with_coord=True, padding_mode="replicate" )( inputs, network,