Skip to content

Commit 13ee1d7

Browse files
committed
Tweak comments
1 parent cd15a7b commit 13ee1d7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/open_clip/factory.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,9 @@
2929
try:
3030
import _codecs
3131
import numpy as np
32-
# add safe globals, known to be needed for metaclip weights
32+
# add safe globals that are known to be needed for metaclip weights loading in weights_only=True mode
3333
torch.serialization.add_safe_globals([
34-
_codecs.encode, # now in pytorch main but some pytorch versions w/ weights_only flag don't have it
34+
_codecs.encode, # this one not needed for PyTorch >= 2.5.0
3535
np.core.multiarray.scalar,
3636
np.dtype,
3737
np.dtypes.Float64DType,

0 commit comments

Comments
 (0)