Spaces:
Running
on
L40S
Running
on
L40S
update
Browse files- 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(
|
|
|
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 |
|