This commit is contained in:
dc-dc-dc 2023-12-18 22:12:42 -05:00
parent c88d3174aa
commit 99b9c1dac5

View File

@ -2971,7 +2971,7 @@ void init_ops(py::module_& m) {
py::pos_only(),
py::kw_only(),
R"pbdoc(
save_safetensor(file: str, d: Dict[str, array], /, *, stream: Union[None, Stream, Device] = None)
save_safetensor(file: str, d: Dict[str, array], /, *)
Save array(s) to a binary file in ``.safetensors`` format.