Spaces:

fffiloni
/

text-guided-image-colorization

Running on Zero

fffiloni commited on May 28

Commit

fa48cf6

verified ·

1 Parent(s): c78a2af

print accelerator device to locate

Files changed (1) hide show

gradio_ui.py CHANGED Viewed

@@ -201,6 +201,8 @@ def process_image(image_path: str,
         mixed_precision=mixed_precision,
         cpu=False
     )
     weight_dtype = torch.float32
     if accelerator.mixed_precision == "fp16":

         mixed_precision=mixed_precision,
         cpu=False
     )
+    print(f"Accelerator device: {accelerator.device}")
     weight_dtype = torch.float32
     if accelerator.mixed_precision == "fp16":