yisol
commited on
Commit
•
1b0c672
1
Parent(s):
cfede52
update demo code
Browse files- src/unet_hacked_tryon.py +1 -1
src/unet_hacked_tryon.py
CHANGED
@@ -1238,7 +1238,7 @@ class UNet2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoadersMixin)
|
|
1238 |
)
|
1239 |
image_embeds = added_cond_kwargs.get("image_embeds")
|
1240 |
# print(image_embeds.shape)
|
1241 |
-
image_embeds = self.encoder_hid_proj(image_embeds).to(encoder_hidden_states.dtype)
|
1242 |
encoder_hidden_states = torch.cat([encoder_hidden_states, image_embeds], dim=1)
|
1243 |
|
1244 |
# 2. pre-process
|
|
|
1238 |
)
|
1239 |
image_embeds = added_cond_kwargs.get("image_embeds")
|
1240 |
# print(image_embeds.shape)
|
1241 |
+
# image_embeds = self.encoder_hid_proj(image_embeds).to(encoder_hidden_states.dtype)
|
1242 |
encoder_hidden_states = torch.cat([encoder_hidden_states, image_embeds], dim=1)
|
1243 |
|
1244 |
# 2. pre-process
|