Update handler.py

handler.py

@@ -157,9 +157,30 @@ class EndpointHandler:
             tmp_vae_path = setup_vae_path(original_vae_path)
             
             # Override the VAE path in constants to use our temporary directory
-            from hyvideo.constants import VAE_PATH
+            from hyvideo.constants import VAE_PATH, TEXT_ENCODER_PATH, TOKENIZER_PATH
             VAE_PATH["884-16c-hy"] = str(tmp_vae_path)
             logger.info(f"Updated VAE_PATH to: {VAE_PATH['884-16c-hy']}")
+            
+            # Update text encoder paths to use absolute paths
+            text_encoder_path = str(Path(path) / "text_encoder")
+            text_encoder_2_path = str(Path(path) / "text_encoder_2")
+            
+            # Update both text encoder and tokenizer paths
+            TEXT_ENCODER_PATH.update({
+                "llm": text_encoder_path,
+                "clipL": text_encoder_2_path
+            })
+            
+            TOKENIZER_PATH.update({
+                "llm": text_encoder_path,
+                "clipL": text_encoder_2_path
+            })
+            
+            logger.info(f"Updated text encoder paths:")
+            logger.info(f"TEXT_ENCODER_PATH['llm']: {TEXT_ENCODER_PATH['llm']}")
+            logger.info(f"TEXT_ENCODER_PATH['clipL']: {TEXT_ENCODER_PATH['clipL']}")
+            logger.info(f"TOKENIZER_PATH['llm']: {TOKENIZER_PATH['llm']}")
+            logger.info(f"TOKENIZER_PATH['clipL']: {TOKENIZER_PATH['clipL']}")
         
         self.args.dit_weight = str(dit_weight_path)
         
@@ -175,7 +196,7 @@ class EndpointHandler:
         except Exception as e:
             logger.error(f"Error initializing model: {str(e)}")
             raise
-    
+        
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
         """Process a single request"""
         # Log incoming request