Spaces:

willsh1997
/

littlefreaksgradio

Running on Zero

willsh1997 commited on Jan 16

Commit

291e403

verified ·

1 Parent(s): 86af578

whoops

Files changed (1) hide show

app.py CHANGED Viewed

@@ -317,7 +317,7 @@ class customUnClipPipeline(UnCLIPImageVariationPipeline):
 torch_device = 'cpu'
 pipe = customUnClipPipeline.from_pretrained("kakaobrain/karlo-v1-alpha-image-variations", torch_dtype=torch.float32, trust_remote_code=True,
                                             # accelerator='ort', device=torch_device
-                                            device_map='cpu'
                                            )
 # pipe.to(torch_device)
 # pipe.enable_model_cpu_offload()

 torch_device = 'cpu'
 pipe = customUnClipPipeline.from_pretrained("kakaobrain/karlo-v1-alpha-image-variations", torch_dtype=torch.float32, trust_remote_code=True,
                                             # accelerator='ort', device=torch_device
+                                            # device_map='cpu'
                                            )
 # pipe.to(torch_device)
 # pipe.enable_model_cpu_offload()