JingyeChen22
commited on
Commit
·
80655d1
1
Parent(s):
774205b
Update app.py
Browse files
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 |
|