autoencoder

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.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]