fix laion.py imports

2023-12-25 20:42:21 -05:00
parent 8b76026b45
commit 40d4e5155c
1 changed files with 1 additions and 1 deletions
@@ -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()