• brkirch's avatar
    Newer versions of PyTorch use TypedStorage instead · 8acc901b
    brkirch authored
    Pytorch 1.13 and later will rename _TypedStorage to TypedStorage, so check for TypedStorage and use _TypedStorage if it is not available. Currently this is needed so that nightly builds of PyTorch work correctly.
    8acc901b
safe.py 3.34 KB