We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6491051 commit c44ab44Copy full SHA for c44ab44
lightx2v/utils/utils.py
@@ -6,6 +6,7 @@
6
import imageio
7
import imageio_ffmpeg as ffmpeg
8
import numpy as np
9
+import safetensors
10
import torch
11
import torch.distributed as dist
12
import torchvision
@@ -15,7 +16,6 @@
15
16
from torchvision.transforms import InterpolationMode
17
from torchvision.transforms.functional import resize
18
-import safetensors
19
from lightx2v_platform.base.global_var import AI_DEVICE
20
21
torch_device_module = getattr(torch, AI_DEVICE)
0 commit comments