YulianSa commited on
Commit
021ce54
·
1 Parent(s): a6df575
Files changed (1) hide show
  1. infer_api.py +2 -2
infer_api.py CHANGED
@@ -766,9 +766,9 @@ infer_multiview_pipeline = StableUnCLIPImg2ImgPipeline.from_pretrained(
766
  infer_multiview_cfg.pretrained_path,
767
  torch_dtype=torch.float16,)
768
  if torch.cuda.is_available():
769
- infer_multiview_pipeline.to(device)
 
770
  infer_multiview_pipeline.unet.enable_xformers_memory_efficient_attention()
771
- print(f"Era3D Using device!!!!!!!!!!!!: {device}", file=sys.stderr)
772
 
773
  infer_multiview_results = {}
774
 
 
766
  infer_multiview_cfg.pretrained_path,
767
  torch_dtype=torch.float16,)
768
  if torch.cuda.is_available():
769
+ infer_multiview_pipeline.to(infer_multiview_device)
770
+ print(f"Era3D Using device!!!!!!!!!!!!: {infer_multiview_device}", file=sys.stderr)
771
  infer_multiview_pipeline.unet.enable_xformers_memory_efficient_attention()
 
772
 
773
  infer_multiview_results = {}
774