Westlake-AGI-Lab commited on
Commit
2993b8e
·
verified ·
1 Parent(s): dae85a6

Update utils/ptp_utils.py

Browse files
Files changed (1) hide show
  1. utils/ptp_utils.py +0 -2
utils/ptp_utils.py CHANGED
@@ -1,7 +1,6 @@
1
  import cv2
2
  import numpy as np
3
  import torch
4
- from IPython.display import display
5
  from PIL import Image
6
  from einops import rearrange
7
  import math
@@ -84,4 +83,3 @@ def view_images(images, num_rows=1, offset_ratio=0.02):
84
  i * num_cols + j]
85
 
86
  pil_img = Image.fromarray(image_)
87
- display(pil_img)
 
1
  import cv2
2
  import numpy as np
3
  import torch
 
4
  from PIL import Image
5
  from einops import rearrange
6
  import math
 
83
  i * num_cols + j]
84
 
85
  pil_img = Image.fromarray(image_)