Spaces:
Running
on
Zero
Running
on
Zero
parokshsaxena
commited on
Commit
β’
08a2700
1
Parent(s):
3f3bc41
printing input
Browse files- app.py +1 -0
- 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:
|