Spaces:

fffiloni
/

text-guided-image-colorization

Running on Zero

fffiloni commited on 23 days ago

Commit

87d5ed1

verified ·

1 Parent(s): 8497704

Update gradio_ui.py

Files changed (1) hide show

gradio_ui.py CHANGED Viewed

@@ -199,8 +199,7 @@ def process_image(image_path: str,
     # Accelerator Setting
     accelerator = Accelerator(
         mixed_precision=mixed_precision,
-        device="cuda"
-    )
     weight_dtype = torch.float32
     if accelerator.mixed_precision == "fp16":

     # Accelerator Setting
     accelerator = Accelerator(
         mixed_precision=mixed_precision,
+    ).to("cuda")
     weight_dtype = torch.float32
     if accelerator.mixed_precision == "fp16":