wanghaofan commited on
Commit
f0394a9
1 Parent(s): d11a495

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -63,6 +63,10 @@ strength_mapping = {"canny":0.65, "tile":0.45, "depth":0.55, "blur":0.45, "openp
63
  canny = CannyDetector()
64
  open_pose = OpenposeDetector.from_pretrained("lllyasviel/Annotators")
65
 
 
 
 
 
66
  def convert_from_image_to_cv2(img: Image) -> np.ndarray:
67
  return cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
68
 
 
63
  canny = CannyDetector()
64
  open_pose = OpenposeDetector.from_pretrained("lllyasviel/Annotators")
65
 
66
+ torch.backends.cuda.matmul.allow_tf32 = True
67
+ pipe.vae.enable_tiling()
68
+ pipe.vae.enable_slicing()
69
+
70
  def convert_from_image_to_cv2(img: Image) -> np.ndarray:
71
  return cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
72