Merge pull request #283 from misha-chertushkin/master

Add safetensors loading from local environment
2025-07-07 15:12:13 -07:00
parent 95bdf54d50 6cf4f23675
commit ba034ae71c
2 changed files with 27 additions and 11 deletions
@@ -39,6 +39,7 @@ scikit-learn = ">=1.2.2"
 typer = ">=0.12.3"
 wandb = ">=0.17.5"
 absl-py = ">=1.4.0"
+safetensors = "^0.5.3"

 [tool.poetry.extras]
 pax = ["paxml", "lingvo", "jax", "jaxlib"]