Spaces:

willsh1997
/

littlefreaksgradio

Running on Zero

willsh1997 commited on Jan 16

Commit

d45dbc3

verified ·

1 Parent(s): 291e403

move pipeline back onto cuda

Files changed (1) hide show

app.py CHANGED Viewed

@@ -319,7 +319,7 @@ pipe = customUnClipPipeline.from_pretrained("kakaobrain/karlo-v1-alpha-image-var
                                             # accelerator='ort', device=torch_device
                                             # device_map='cpu'
                                            )
-# pipe.to(torch_device)
 # pipe.enable_model_cpu_offload()

                                             # accelerator='ort', device=torch_device
                                             # device_map='cpu'
                                            )
+pipe.to("cuda")
 # pipe.enable_model_cpu_offload()