File size: 135 Bytes
5fa1a76
1
If you want the AutoModel API to cast the load the checkpoints with the storage weights type, you must specify torch_dtype="auto", e.g.