Spaces:

fffiloni
/

text-guided-image-colorization

Running on Zero

fffiloni commited on 15 days ago

Commit

c78a2af

verified ·

1 Parent(s): 87d5ed1

force gpu usage for Accelerator according to its init attributes

Files changed (1) hide show

gradio_ui.py CHANGED Viewed

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

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