Spaces:

tttoaster
/

SEED-X-17B

Build error

tttoaster commited on May 15, 2024

Commit

bfa897b

verified ·

1 Parent(s): 25c55b9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -368,7 +368,7 @@ def generate(text_list, image_list, max_new_tokens, force_boi, force_bbox, force
                 torch.cuda.empty_cache()
-                service.vae_pipe = service.vae_pipe.to(self.vit_sd_device)
                 init_image = generated_image.resize((1024, 1024))
                 prompt = ""

                 torch.cuda.empty_cache()
+                service.vae_pipe = service.vae_pipe.to(service.vit_sd_device)
                 init_image = generated_image.resize((1024, 1024))
                 prompt = ""