Fix TP regression

Author: turboderp

exllamav3/model/model_tp.py

@@ -353,7 +353,7 @@ def prepare_inputs_for_tp(self, x: torch.Tensor, params: dict) -> torch.Tensor:
             "cache_seqlens",
             "positions",
             "position_ids",
-            "indexed_embeddings",
+            # "indexed_embeddings",
         ]:
             p = params.get(tensor_param)
             if p is not None:

exllamav3/model/model_tp_fn.py

@@ -191,7 +191,7 @@ def mp_model_forward(
         "cache_seqlens",
         "positions",
         "position_ids",
-        "indexed_embeddings",
+        # "indexed_embeddings",
     ]:
         p = params.get(tensor_param)
         if p is not None: