diff --git a/classifiers/laion.py b/classifiers/laion.py
index d42a90c..51a23b2 100644
--- a/classifiers/laion.py
+++ b/classifiers/laion.py
@@ -3,7 +3,7 @@ import pathlib
 import numpy as np
 import clip
 import torch
-import safetensors
+import safetensors.torch
 
 use_cuda = torch.cuda.is_available()