Skip to content

Commit a4bec95

Browse files
cpgaffney1Orbax Authors
cpgaffney1
authored and
Orbax Authors
committed
#v1 #safetensors Refactor SafetensorsLayout to eliminate unperformant multi-host loading code. On a single-host, a single read operation is performed, followed by device_put to local devices.
PiperOrigin-RevId: 893168206
1 parent 39262e1 commit a4bec95

2 files changed

Lines changed: 160 additions & 254 deletions

File tree

0 commit comments

Comments
 (0)