Spaces:

ELITE-library
/

ELITE

Sleeping

csyxwei commited on Jul 4

Commit

0fc5706

•

1 Parent(s): df2388c

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -244,7 +244,8 @@ class Model:
         lambda_: float,
         num_steps: int,
     ) -> PIL.Image.Image:
-        data = self.prepare_data(image['image'], image['mask'], text)
         uncond_input = self.tokenizer(
             [''] * data['pixel_values'].shape[0],

         lambda_: float,
         num_steps: int,
     ) -> PIL.Image.Image:
+        # data = self.prepare_data(image['image'], image['mask'], text)
+        data = self.prepare_data(image['background'], image['layers'][0], text)
         uncond_input = self.tokenizer(
             [''] * data['pixel_values'].shape[0],