Update README.md
Browse files
README.md
CHANGED
|
@@ -50,10 +50,12 @@ All models were created and saved using the official Ultralytics library, so it
|
|
| 50 |
See also: https://huggingface.co/docs/hub/security-pickle
|
| 51 |
|
| 52 |
**Note on Loading Weights**
|
|
|
|
| 53 |
If you encounter an error like:
|
| 54 |
```bash
|
| 55 |
Weights-only load failed. Unsupported global: builtins.set
|
| 56 |
```
|
|
|
|
| 57 |
This happens because PyTorch (for safety) blocks loading some objects by default.
|
| 58 |
To fix it safely, add this code before loading:
|
| 59 |
```bash
|
|
@@ -67,4 +69,4 @@ This whitelists set and allows safe loading with weights_only=True.
|
|
| 67 |
|
| 68 |
Otherwise, never disable weights_only unless you 100% trust the file.
|
| 69 |
|
| 70 |
-
Check
|
|
|
|
| 50 |
See also: https://huggingface.co/docs/hub/security-pickle
|
| 51 |
|
| 52 |
**Note on Loading Weights**
|
| 53 |
+
|
| 54 |
If you encounter an error like:
|
| 55 |
```bash
|
| 56 |
Weights-only load failed. Unsupported global: builtins.set
|
| 57 |
```
|
| 58 |
+
|
| 59 |
This happens because PyTorch (for safety) blocks loading some objects by default.
|
| 60 |
To fix it safely, add this code before loading:
|
| 61 |
```bash
|
|
|
|
| 69 |
|
| 70 |
Otherwise, never disable weights_only unless you 100% trust the file.
|
| 71 |
|
| 72 |
+
Check [PyTorch's torch.load docs](https://pytorch.org/docs/stable/generated/torch.load.html) for more info.
|