autodecoder

app.py

@@ -135,21 +135,21 @@ clip = FrozenCLIPEmbedder()
 clip.eval()
 clip.to(device)
 
-# # Load autoencoder.
-# autoencoder = libs.autoencoder.get_model(**config.autoencoder)
-# autoencoder.to(device)
+# Load autoencoder.
+autoencoder = libs.autoencoder.get_model(**config_1.autoencoder)
+autoencoder.to(device)
 
 
-# @torch.cuda.amp.autocast()
-# def encode(_batch: torch.Tensor) -> torch.Tensor:
-#     """Encode a batch of images using the autoencoder."""
-#     return autoencoder.encode(_batch)
+@torch.cuda.amp.autocast()
+def encode(_batch: torch.Tensor) -> torch.Tensor:
+    """Encode a batch of images using the autoencoder."""
+    return autoencoder.encode(_batch)
 
 
-# @torch.cuda.amp.autocast()
-# def decode(_batch: torch.Tensor) -> torch.Tensor:
-#     """Decode a batch of latent vectors using the autoencoder."""
-#     return autoencoder.decode(_batch)
+@torch.cuda.amp.autocast()
+def decode(_batch: torch.Tensor) -> torch.Tensor:
+    """Decode a batch of latent vectors using the autoencoder."""
+    return autoencoder.decode(_batch)
 
 
 @spaces.GPU #[uncomment to use ZeroGPU]