Carlos s commited on
Commit
11450b9
·
verified ·
1 Parent(s): 248af78

Update LTX-Video/ltx_video/models/autoencoders/causal_video_autoencoder.py

Browse files
LTX-Video/ltx_video/models/autoencoders/causal_video_autoencoder.py CHANGED
@@ -556,12 +556,14 @@ class Encoder(nn.Module):
556
  dim=1,
557
  )
558
 
559
-
 
 
560
 
561
  # --- INÍCIO DO PATCH CIRÚRGICO ADUC ---
562
  # Verificamos uma variável de ambiente para ligar/desligar o overlay
563
 
564
- if os.getenv("ADUC_DEBUG_OVERLAY", "1") == "1":
565
  try:
566
  print(f"[ADUC DEBUG LTX *causal_video_autoencoder.py*]=======")
567
  print(f"[sample] {sample.shape}")
@@ -847,6 +849,12 @@ class Decoder(nn.Module):
847
 
848
  sample = unpatchify(sample, patch_size_hw=self.patch_size, patch_size_t=1)
849
 
 
 
 
 
 
 
850
  return sample
851
 
852
 
 
556
  dim=1,
557
  )
558
 
559
+ print(f"[ADUC DEBUG LTX *causal_video_autoencoder.py*]=======")
560
+ print(f"[sample][decode] {sample.shape}")
561
+
562
 
563
  # --- INÍCIO DO PATCH CIRÚRGICO ADUC ---
564
  # Verificamos uma variável de ambiente para ligar/desligar o overlay
565
 
566
+ if os.getenv("ADUC_DEBUG_OVERLAY", "1") == "2":
567
  try:
568
  print(f"[ADUC DEBUG LTX *causal_video_autoencoder.py*]=======")
569
  print(f"[sample] {sample.shape}")
 
849
 
850
  sample = unpatchify(sample, patch_size_hw=self.patch_size, patch_size_t=1)
851
 
852
+
853
+ print(f"[ADUC DEBUG LTX *causal_video_autoencoder.py*]=======")
854
+ print(f"[sample][encode] {sample.shape}")
855
+
856
+
857
+
858
  return sample
859
 
860