parokshsaxena commited on
Commit
08a2700
β€’
1 Parent(s): 3f3bc41

printing input

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. src/tryon_pipeline.py +2 -2
app.py CHANGED
@@ -154,6 +154,7 @@ def is_cropping_required(width, height):
154
  @spaces.GPU
155
  def start_tryon(human_img_dict,garm_img,garment_des, background_img, is_checked,is_checked_crop,denoise_steps,seed):
156
  logging.info("Starting try on")
 
157
  #device = "cuda"
158
  device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
159
 
 
154
  @spaces.GPU
155
  def start_tryon(human_img_dict,garm_img,garment_des, background_img, is_checked,is_checked_crop,denoise_steps,seed):
156
  logging.info("Starting try on")
157
+ print(f"Input: {human_img_dict}")
158
  #device = "cuda"
159
  device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
160
 
src/tryon_pipeline.py CHANGED
@@ -1791,8 +1791,8 @@ class StableDiffusionXLInpaintPipeline(
1791
  # down,reference_features = self.UNet_Encoder(cloth,t, text_embeds_cloth,added_cond_kwargs= {"text_embeds": pooled_prompt_embeds_c, "time_ids": add_time_ids},return_dict=False)
1792
  down,reference_features = self.unet_encoder(cloth,t, text_embeds_cloth,return_dict=False)
1793
  # garment_out, reference_features = self.garment_net(cloth, t, text_embeds_cloth)
1794
- print(type(reference_features))
1795
- print(reference_features)
1796
  reference_features = list(reference_features)
1797
  # print(len(reference_features))
1798
  # for elem in reference_features:
 
1791
  # down,reference_features = self.UNet_Encoder(cloth,t, text_embeds_cloth,added_cond_kwargs= {"text_embeds": pooled_prompt_embeds_c, "time_ids": add_time_ids},return_dict=False)
1792
  down,reference_features = self.unet_encoder(cloth,t, text_embeds_cloth,return_dict=False)
1793
  # garment_out, reference_features = self.garment_net(cloth, t, text_embeds_cloth)
1794
+ #print(type(reference_features))
1795
+ #print(reference_features)
1796
  reference_features = list(reference_features)
1797
  # print(len(reference_features))
1798
  # for elem in reference_features: