Update gradio_app.py

gradio_app.py

@@ -154,8 +154,8 @@ def gen_mvimg(
         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")
         
-        imgs_in = imgs_in.to(device=device, dtype=torch.float16)
-        prompt_embeddings = prompt_embeddings.to(device=device, dtype=torch.float16)
+        imgs_in = imgs_in.to(dtype=torch.float16)
+        prompt_embeddings = prompt_embeddings.to(dtype=torch.float16)
         
         mv_imgs = era3d_pipeline(
             imgs_in, 
@@ -259,7 +259,7 @@ if __name__=="__main__":
         )
         # enable xformers
         # era3d_pipeline.unet.enable_xformers_memory_efficient_attention()
-        era3d_pipeline.to(device)
+        # era3d_pipeline.to(device)
     elif "CRM" in mvimg_model_config_list:
         stage1_config = OmegaConf.load(f"apps/third_party/CRM/configs/nf7_v3_SNR_rd_size_stroke.yaml").config
         stage1_sampler_config = stage1_config.sampler