Spaces:

willsh1997
/

littlefreaksgradio

Running on Zero

willsh1997 commited on Jan 16

Commit

1c9ad26

verified ·

1 Parent(s): 3a43f8c

delete cpu code

Files changed (1) hide show

app.py CHANGED Viewed

@@ -314,7 +314,9 @@ class customUnClipPipeline(UnCLIPImageVariationPipeline):
 ### ADDITIONAL PIPELINE CODE FOR KARLO
-pipe = customUnClipPipeline.from_pretrained("kakaobrain/karlo-v1-alpha-image-variations", torch_dtype=torch.float32, trust_remote_code=True, accelerator='ort', device='cpu')
 # pipe.enable_model_cpu_offload()

 ### ADDITIONAL PIPELINE CODE FOR KARLO
+pipe = customUnClipPipeline.from_pretrained("kakaobrain/karlo-v1-alpha-image-variations", torch_dtype=torch.float32, trust_remote_code=True,
+                                            # accelerator='ort', device='cpu'
+                                           )
 # pipe.enable_model_cpu_offload()