csyxwei commited on
Commit
0fc5706
1 Parent(s): df2388c

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -1
model.py CHANGED
@@ -244,7 +244,8 @@ class Model:
244
  lambda_: float,
245
  num_steps: int,
246
  ) -> PIL.Image.Image:
247
- data = self.prepare_data(image['image'], image['mask'], text)
 
248
 
249
  uncond_input = self.tokenizer(
250
  [''] * data['pixel_values'].shape[0],
 
244
  lambda_: float,
245
  num_steps: int,
246
  ) -> PIL.Image.Image:
247
+ # data = self.prepare_data(image['image'], image['mask'], text)
248
+ data = self.prepare_data(image['background'], image['layers'][0], text)
249
 
250
  uncond_input = self.tokenizer(
251
  [''] * data['pixel_values'].shape[0],