fix laion.py imports
This commit is contained in:
@@ -3,7 +3,7 @@ import pathlib
|
|||||||
import numpy as np
|
import numpy as np
|
||||||
import clip
|
import clip
|
||||||
import torch
|
import torch
|
||||||
import safetensors
|
import safetensors.torch
|
||||||
|
|
||||||
use_cuda = torch.cuda.is_available()
|
use_cuda = torch.cuda.is_available()
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user