Spaces:
Paused
Paused
Update managers/vae_manager.py
Browse files- managers/vae_manager.py +3 -0
managers/vae_manager.py
CHANGED
|
@@ -4,6 +4,9 @@
|
|
| 4 |
import torch
|
| 5 |
import contextlib
|
| 6 |
|
|
|
|
|
|
|
|
|
|
| 7 |
class _SimpleVAEManager:
|
| 8 |
def __init__(self, pipeline=None, device=None, autocast_dtype=torch.float32):
|
| 9 |
"""
|
|
|
|
| 4 |
import torch
|
| 5 |
import contextlib
|
| 6 |
|
| 7 |
+
from ltx_video.models.autoencoders.vae_encode import vae_encode, vae_decode
|
| 8 |
+
|
| 9 |
+
|
| 10 |
class _SimpleVAEManager:
|
| 11 |
def __init__(self, pipeline=None, device=None, autocast_dtype=torch.float32):
|
| 12 |
"""
|