diff --git a/README.md b/README.md
index 4902861..6371f50 100644
--- a/README.md
+++ b/README.md
@@ -17,7 +17,7 @@ This is still work in progress, like everything else.
 ## Javascript
 
 ### browserstatus.js
-Sets the favicon to green circle when not processing anything, sets it to red when processing and shows progress percentage and the lenghth of your queue. 
+Sets the favicon to green circle when not processing anything, sets it to red when processing and shows progress percentage and the length of your queue. 
 Default off, needs to be enabled from options, overrides Custom-Scripts favicon when enabled.
 
 ## Nodes:
diff --git a/nodes/model_optimization_nodes.py b/nodes/model_optimization_nodes.py
index 556f87d..2aefb39 100644
--- a/nodes/model_optimization_nodes.py
+++ b/nodes/model_optimization_nodes.py
@@ -375,15 +375,41 @@ class TorchCompileVAE:
     def compile(self, vae, backend, mode, fullgraph, compile_encoder, compile_decoder):
         if compile_encoder:
             if not self._compiled_encoder:
+                encoder_name = "encoder"
+                if hasattr(vae.first_stage_model, "taesd_encoder"):
+                    encoder_name = "taesd_encoder"
+
                 try:
-                    vae.first_stage_model.encoder = torch.compile(vae.first_stage_model.encoder, mode=mode, fullgraph=fullgraph, backend=backend)
+                    setattr(
+                        vae.first_stage_model,
+                        encoder_name,
+                        torch.compile(
+                            getattr(vae.first_stage_model, encoder_name),
+                            mode=mode,
+                            fullgraph=fullgraph,
+                            backend=backend,
+                        ),
+                    )
                     self._compiled_encoder = True
                 except:
                     raise RuntimeError("Failed to compile model")
         if compile_decoder:
             if not self._compiled_decoder:
+                decoder_name = "decoder"
+                if hasattr(vae.first_stage_model, "taesd_decoder"):
+                    decoder_name = "taesd_decoder"
+
                 try:
-                    vae.first_stage_model.decoder = torch.compile(vae.first_stage_model.decoder, mode=mode, fullgraph=fullgraph, backend=backend)
+                    setattr(
+                        vae.first_stage_model,
+                        decoder_name,
+                        torch.compile(
+                            getattr(vae.first_stage_model, decoder_name),
+                            mode=mode,
+                            fullgraph=fullgraph,
+                            backend=backend,
+                        ),
+                    )
                     self._compiled_decoder = True
                 except:
                     raise RuntimeError("Failed to compile model")