Spaces:

fffiloni
/

text-guided-image-colorization

Running on Zero

fffiloni commited on May 28

Commit

8497704

verified ·

1 Parent(s): 3c116d0

set accelerator device to cuda

Files changed (1) hide show

gradio_ui.py CHANGED Viewed

@@ -199,6 +199,7 @@ def process_image(image_path: str,
     # Accelerator Setting
     accelerator = Accelerator(
         mixed_precision=mixed_precision,
     )
     weight_dtype = torch.float32

     # Accelerator Setting
     accelerator = Accelerator(
         mixed_precision=mixed_precision,
+        device="cuda"
     )
     weight_dtype = torch.float32