Yes I have the same problem. In the logs there is a line that indicates that some extensions are ignored when fetching the repository.
INFO | Ignore regex pattern for files, which are not downloaded: onnx , tar.gz, rust , *mlmodel, tflite, tf , openvino , ckpt, flax , *safetensors
I don’t know why these file extensions are ignored…
One potential workaround is maybe to save the weights file under another extension in the repo, and then at the start of __init__ change it back to its original extension but haven’t tried it yet.