Skip to content

GitLab

Unverified Commit 411dd37a authored by gushiqiao's avatar gushiqiao Committed by GitHub
Browse files

fix bug (#346) 


Co-authored-by: default avatarYang Yong (雍洋) <yongyang1030@163.com>
parent 3c464853
Show whitespace changes
Inline Side-by-side
Showing with 8 additions and 1 deletion
lightx2v/models/runners/wan/wan_animate_runner.py
View file @ 411dd37a
...@@ -5,7 +5,14 @@ import cv2 ...@@ -5,7 +5,14 @@ import cv2
import numpy as np import numpy as np
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
from decord import VideoReader from loguru import logger
try:
from decord import VideoReader
except ImportError:
VideoReader = None
logger.info("If you need run animate model, please install decord.")
from lightx2v.models.input_encoders.hf.animate.face_encoder import FaceEncoder from lightx2v.models.input_encoders.hf.animate.face_encoder import FaceEncoder
from lightx2v.models.input_encoders.hf.animate.motion_encoder import Generator from lightx2v.models.input_encoders.hf.animate.motion_encoder import Generator
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment