valhalla commited on
Commit
1878ed0
1 Parent(s): 3c50bd9

fix ZoePreprocessor

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -76,7 +76,7 @@ class ZoePreprocessor:
76
  ).to(device)
77
 
78
  def __call__(self, image: PIL.Image.Image) -> PIL.Image.Image:
79
- return self.model(image, gamma_corrected=True)
80
 
81
 
82
  def get_preprocessor(adapter_name: str) -> Callable[[PIL.Image.Image], PIL.Image.Image]:
 
76
  ).to(device)
77
 
78
  def __call__(self, image: PIL.Image.Image) -> PIL.Image.Image:
79
+ return self.model(image, gamma_corrected=True, image_resolution=1024)
80
 
81
 
82
  def get_preprocessor(adapter_name: str) -> Callable[[PIL.Image.Image], PIL.Image.Image]: