yisol commited on
Commit
1b0c672
1 Parent(s): cfede52

update demo code

Browse files
Files changed (1) hide show
  1. 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