Fix fused params for st publish umia hstu model (#3461)

Summary:
Pull Request resolved: #3461

As title, for st publish, we need to pass tbe config into sharding model.

Reviewed By: bobbyliujb, emlin

Differential Revision: D79698284

fbshipit-source-id: 8236ab2df8c36c9b735e4c239d48b08f677ebe95

Author: EddyLXJ

torchrec/distributed/batched_embedding_kernel.py

@@ -157,6 +157,7 @@ def _populate_ssd_tbe_params(config: GroupedEmbeddingConfig) -> Dict[str, Any]:
     Construct SSD TBE params dict from config and fused params dict.
     """
     fused_params = config.fused_params or {}
+    logger.info(f"Populate_ssd_tbe_params with {fused_params=}")
 
     ssd_tbe_params: Dict[str, Any] = {}