From 40d4e5155c3ebf4ec5241a3cdeb0b4642cacd70e Mon Sep 17 00:00:00 2001
From: Mackerel <mackerel@chickatrice.com>
Date: Mon, 25 Dec 2023 20:42:21 -0500
Subject: [PATCH] fix laion.py imports

---
 classifiers/laion.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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()