diff options
Diffstat (limited to 'candle-examples/examples/whisper/extract_weights.py')
| -rw-r--r-- | candle-examples/examples/whisper/extract_weights.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/whisper/extract_weights.py b/candle-examples/examples/whisper/extract_weights.py index d6ccffc6..65602703 100644 --- a/candle-examples/examples/whisper/extract_weights.py +++ b/candle-examples/examples/whisper/extract_weights.py @@ -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"]) |