Skip to content

Commit

Permalink
add import line
Browse files Browse the repository at this point in the history
  • Loading branch information
Wauplin committed Jan 4, 2024
1 parent 72e6295 commit 9dd1bad
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions docs/source/metadata_parsing.mdx
Original file line number Diff line number Diff line change
Expand Up @@ -97,6 +97,8 @@ Use [`get_safetensors_metadata`](https://huggingface.co/docs/huggingface_hub/pac
Depending on if the model is sharded or not, one or multiple safetensors files will be parsed.

```python
>>> from huggingface_hub import get_safetensors_metadata

# Parse repo with single weights file
>>> metadata = get_safetensors_metadata("bigscience/bloomz-560m")
>>> metadata
Expand Down

0 comments on commit 9dd1bad

Please sign in to comment.