You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Reproduced a simple project with the main.rs from candle-examples/examples/distilbert with minimal dependencies.
Running:
cargo run --release -- --use-pth --prompt "..."
Tries to load pytorch_model.bin and fails with the following error:
invalid Zip archive: Could not find central directory end
Using model.safetensors works, though. I suppose that the PyTorch format of this quite ancient model file (distilbert-base-uncased) is not what is expected (not in ZIP file format).
Is there a way to deal with this using Candle ?
The text was updated successfully, but these errors were encountered:
fbilhaut
changed the title
distilbert example not working when using PyTorch model
The "distilbert" example fails to load the PyTorch model
Dec 13, 2024
Reproduced a simple project with the
main.rs
fromcandle-examples/examples/distilbert
with minimal dependencies.Running:
cargo run --release -- --use-pth --prompt "..."
Tries to load
pytorch_model.bin
and fails with the following error:Using
model.safetensors
works, though. I suppose that the PyTorch format of this quite ancient model file (distilbert-base-uncased
) is not what is expected (not in ZIP file format).Is there a way to deal with this using Candle ?
The text was updated successfully, but these errors were encountered: