Spaces:
Running
on
Zero
Running
on
Zero
print accelerator device to locate
Browse files- gradio_ui.py +2 -0
gradio_ui.py
CHANGED
@@ -201,6 +201,8 @@ def process_image(image_path: str,
|
|
201 |
mixed_precision=mixed_precision,
|
202 |
cpu=False
|
203 |
)
|
|
|
|
|
204 |
|
205 |
weight_dtype = torch.float32
|
206 |
if accelerator.mixed_precision == "fp16":
|
|
|
201 |
mixed_precision=mixed_precision,
|
202 |
cpu=False
|
203 |
)
|
204 |
+
|
205 |
+
print(f"Accelerator device: {accelerator.device}")
|
206 |
|
207 |
weight_dtype = torch.float32
|
208 |
if accelerator.mixed_precision == "fp16":
|