JingyeChen22 commited on
Commit
80655d1
·
1 Parent(s): 774205b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -121,8 +121,8 @@ def exe_redo(i, orig_i, t, guest_id):
121
 
122
  def get_pixels(i, orig_i, radio, t, guest_id, evt: gr.SelectData):
123
 
124
- print('hi1 ', i)
125
- print('hi2 ', orig_i)
126
 
127
  width, height = Image.open(i).size
128
 
@@ -528,12 +528,12 @@ def text_to_image(guest_id, i, orig_i, prompt,keywords,positive_prompt,radio,sli
528
 
529
  image = Image.open(orig_i).resize((512,512))
530
  image_tensor = to_tensor(image).unsqueeze(0).cuda().sub_(0.5).div_(0.5)
531
- print(f'image_tensor.shape {image_tensor.shape}')
532
  masked_image = image_tensor * (1-image_mask)
533
  masked_feature = vae.encode(masked_image.half()).latent_dist.sample()
534
  masked_feature = masked_feature * vae.config.scaling_factor
535
  masked_feature = masked_feature.half()
536
- print(f'masked_feature.shape {masked_feature.shape}')
537
 
538
  feature_mask = torch.nn.functional.interpolate(image_mask, size=(64,64), mode='nearest').cuda()
539
 
 
121
 
122
  def get_pixels(i, orig_i, radio, t, guest_id, evt: gr.SelectData):
123
 
124
+ # print('hi1 ', i)
125
+ # print('hi2 ', orig_i)
126
 
127
  width, height = Image.open(i).size
128
 
 
528
 
529
  image = Image.open(orig_i).resize((512,512))
530
  image_tensor = to_tensor(image).unsqueeze(0).cuda().sub_(0.5).div_(0.5)
531
+ # print(f'image_tensor.shape {image_tensor.shape}')
532
  masked_image = image_tensor * (1-image_mask)
533
  masked_feature = vae.encode(masked_image.half()).latent_dist.sample()
534
  masked_feature = masked_feature * vae.config.scaling_factor
535
  masked_feature = masked_feature.half()
536
+ # print(f'masked_feature.shape {masked_feature.shape}')
537
 
538
  feature_mask = torch.nn.functional.interpolate(image_mask, size=(64,64), mode='nearest').cuda()
539