HuiZhang commited on
Commit
676c130
·
verified ·
1 Parent(s): 6aed6eb

Update src/models/transformer_sd3_SiamLayout.py

Browse files
src/models/transformer_sd3_SiamLayout.py CHANGED
@@ -423,7 +423,7 @@ class SiamLayoutSD3Transformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin,
423
 
424
  height, width = hidden_states.shape[-2:]
425
 
426
- hidden_states = self.pos_embed(hidden_states) # takes care of adding positional embeddings too.
427
  temb = self.time_text_embed(timestep, pooled_projections)
428
  encoder_hidden_states = self.context_embedder(encoder_hidden_states)
429
 
 
423
 
424
  height, width = hidden_states.shape[-2:]
425
 
426
+ hidden_states = self.pos_embed(hidden_states)
427
  temb = self.time_text_embed(timestep, pooled_projections)
428
  encoder_hidden_states = self.context_embedder(encoder_hidden_states)
429