hysts HF staff valhalla commited on
Commit
15680c5
1 Parent(s): 26721d3

set lineart detect_resolution to 384 (#3)

Browse files

- set lineart detect_resolution to 384 (b5441a4871561c3b7d955db24bbd39ef049f6541)


Co-authored-by: Suraj Patil <[email protected]>

Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -117,7 +117,7 @@ class LineartPreprocessor(Preprocessor):
117
  return self.model.to(device)
118
 
119
  def __call__(self, image: PIL.Image.Image) -> PIL.Image.Image:
120
- return self.model(image, detect_resolution=512, image_resolution=1024)
121
 
122
 
123
  class MidasPreprocessor(Preprocessor):
 
117
  return self.model.to(device)
118
 
119
  def __call__(self, image: PIL.Image.Image) -> PIL.Image.Image:
120
+ return self.model(image, detect_resolution=384, image_resolution=1024)
121
 
122
 
123
  class MidasPreprocessor(Preprocessor):