add some logs for david

2026-02-24 12:12:38 +00:00
parent 6e86e0386e
commit 93a0f5b7d5
1 changed files with 1 additions and 1 deletions
@@ -623,7 +623,7 @@ class DeepSeekShardingStrategy(TensorParallelShardingStrategy):
            def log_info(stuff: nn.Module, name: str):
                logger.info(f"Info for {name}:")

-                weights = stuff.weights
+                weights = stuff.weight
                logger.info(f"Scales: {weights.shape} {weights.dtype}")

                if hasattr(stuff, "scales"):