Upload folder using huggingface_hub

modeling_sd_gray_inpaint.py

@@ -54,7 +54,7 @@ class SDGrayInpaintModel(PreTrainedModel):
         generator = torch.Generator()
         generator.manual_seed(seed)
         if input_type=='pil':
-            images_gray_masked = self.image_processor.process(images_gray_masked, height=self.height, width=self.width).float()
+            images_gray_masked = self.image_processor.preprocess(images_gray_masked, height=self.height, width=self.width).float()
         elif input_type=='pt':
             images_gray_masked=images_gray_masked
         else:

modeling_seresvae.py

@@ -53,7 +53,7 @@ class SeResVaeModel(PreTrainedModel):
 
     def forward(self, images_gray, input_type='pil', output_type='pil'):
         if input_type=='pil':
-            images_gray = self.image_processor.process(images_gray, height=self.height, width=self.width).float()
+            images_gray = self.image_processor.preprocess(images_gray, height=self.height, width=self.width).float()
         elif input_type=='pt':
             images_gray=images_gray
         else: