feat(megatron): use newer version of tensorizer

harubaru · harubaru · commit 84de79757c7b · 2024-12-02T13:07:34.000-07:00
diff --git a/megatron/requirements.txt b/megatron/requirements.txt
@@ -4,7 +4,7 @@ pyyaml==6.0.2
 regex==2024.9.11
 tensorboard==2.18.0
 tensorboard-data-server==0.7.2
-tensorizer==2.9.0
+tensorizer==2.9.1
 transformers==4.45.2
 triton==3.0.0
 wandb==0.18.3
