Spaces:

bytedance-research
/

RealCustom

Runtime error

CoreloneH commited on Apr 17

Commit

3765927

1 Parent(s): 286a43b

fix cpu debug

Files changed (1) hide show

inference/pipeline.py CHANGED Viewed

@@ -44,7 +44,7 @@ class RealCustomInferencePipeline:
         vae_config="ckpts/sdxl/vae/sdxl.json",
         vae_checkpoint="ckpts/sdxl/vae/sdxl-vae.pth",
         model_type="bf16",
-        device="cuda",
     ):
         if model_type == "bf16":
             self.torch_dtype = torch.bfloat16

         vae_config="ckpts/sdxl/vae/sdxl.json",
         vae_checkpoint="ckpts/sdxl/vae/sdxl-vae.pth",
         model_type="bf16",
+        device="cpu",
     ):
         if model_type == "bf16":
             self.torch_dtype = torch.bfloat16