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
         | 
