guangkaixu commited on
Commit
79891b1
1 Parent(s): 5fbee48
__pycache__/pipeline_genpercept.cpython-310.pyc ADDED
Binary file (7.97 kB). View file
 
pipeline_genpercept.py CHANGED
@@ -152,6 +152,10 @@ class GenPerceptPipeline(DiffusionPipeline):
152
  rgb_norm = rgb_norm[None].to(device)
153
  assert rgb_norm.min() >= -1.0 and rgb_norm.max() <= 1.0
154
  bs_imgs = 1
 
 
 
 
155
 
156
  # ----------------- Predicting depth -----------------
157
 
 
152
  rgb_norm = rgb_norm[None].to(device)
153
  assert rgb_norm.min() >= -1.0 and rgb_norm.max() <= 1.0
154
  bs_imgs = 1
155
+
156
+ print('rgb_norm :', rgb_norm.dtype)
157
+ print('unet :', self.unet.dtype)
158
+ print('vae :', self.vae.dtype)
159
 
160
  # ----------------- Predicting depth -----------------
161
 
util/__pycache__/__init__.cpython-310.pyc ADDED
Binary file (139 Bytes). View file
 
util/__pycache__/batchsize.cpython-310.pyc ADDED
Binary file (1.71 kB). View file
 
util/__pycache__/image_util.cpython-310.pyc ADDED
Binary file (4.8 kB). View file