'update'

app.py

@@ -23,7 +23,7 @@ from einops import rearrange
 import numpy as np
 import subprocess
 from datetime import datetime
-
+from icecream import ic
 def save_image(tensor):
     ndarr = tensor.mul(255).add_(0.5).clamp_(0, 255).permute(1, 2, 0).to("cpu", torch.uint8).numpy()
     # pdb.set_trace()
@@ -186,7 +186,15 @@ def run_pipeline(pipeline, cfg, single_image, guidance_scale, steps, seed, crop_
     prompt_embeddings = torch.cat([normal_prompt_embeddings, clr_prompt_embeddings], dim=0)
     prompt_embeddings = rearrange(prompt_embeddings, "B Nv N C -> (B Nv) N C")
     
-
+    ic(pipeline.unet.device)
+    ic(pipeline.unet.dtype)
+    
+    ic(imgs_in.device)
+    ic(imgs_in.dtype)
+    
+    ic(prompt_embeddings.device)
+    ic(prompt_embeddings.dtype)
+    
     out = pipeline(
         imgs_in, 
         None, 
@@ -296,11 +304,9 @@ def run_demo():
 
     pipeline = load_era3d_pipeline(cfg)
     torch.set_grad_enabled(False)
-    print(1)
     pipeline.to(device=f'cuda:{_GPU_ID}')
     # pipeline.unet.enable_xformers_memory_efficient_attention()
     
-    print(2)
     predictor = sam_init()