huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 02:38:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Run the text decoding bit.

Browse Source
This commit is contained in:
laurent
2023-07-04 12:03:28 +01:00
parent 950b4af49e
commit aea090401d
2 changed files with 24 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-examples/examples/whisper/extract_weights.py
View File

@ -8,6 +8,6 @@ data = torch.load("tiny.en.pt")
weights = {}
for k, v in data["model_state_dict"].items():
weights[k] = v.contiguous()
print(k, v.shape)
print(k, v.shape, v.dtype)
save_file(weights, "tiny.en.safetensors")
print(data["dims"])