Spaces:
Running
on
Zero
Running
on
Zero
HuiZhang
commited on
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)
|
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 |
|