Carlos s
commited on
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") == "
|
| 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 |
|