Skip to content

Commit 16f6b1f

Browse files
authored
Fix logging error
1 parent 06a8f55 commit 16f6b1f

1 file changed

Lines changed: 4 additions & 6 deletions

File tree

src/maxdiffusion/pipelines/ltx_video/ltx_video_pipeline.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -60,12 +60,10 @@
6060

6161
def validate_transformer_inputs(prompt_embeds, fractional_coords, latents, encoder_attention_segment_ids):
6262
# Note: reference shape annotated for first pass default inference parameters
63-
max_logging.log("prompts_embeds.shape: ", prompt_embeds.shape, prompt_embeds.dtype) # (3, 256, 4096) float32
64-
max_logging.log("fractional_coords.shape: ", fractional_coords.shape, fractional_coords.dtype) # (3, 3, 3072) float32
65-
max_logging.log("latents.shape: ", latents.shape, latents.dtype) # (1, 3072, 128) float 32
66-
max_logging.log(
67-
"encoder_attention_segment_ids.shape: ", encoder_attention_segment_ids.shape, encoder_attention_segment_ids.dtype
68-
) # (3, 256) int32
63+
max_logging.log(f"prompts_embeds.shape: {prompt_embeds.shape}") # (3, 256, 4096) float32
64+
max_logging.log(f"fractional_coords.shape: {fractional_coords.shape}") # (3, 3, 3072) float32
65+
max_logging.log(f"latents.shape: {latents.shape}") # (1, 3072, 128) float 32
66+
max_logging.log(f"encoder_attention_segment_ids.shape: {encoder_attention_segment_ids.shape}") # (3, 256) int32
6967

7068

7169
class LTXVideoPipeline:

0 commit comments

Comments
 (0)