Commit 75c03057 authored by helloyongyang's avatar helloyongyang
Browse files

删除t2v和i2v的区分,统一接口

parent efb4d161
......@@ -3,7 +3,7 @@ import numpy as np
from diffusers.utils.torch_utils import randn_tensor
from typing import Union, Tuple, List
from typing import Any, Callable, Dict, List, Optional, Union, Tuple
from lightx2v.text2v.models.schedulers.scheduler import BaseScheduler
from lightx2v.models.schedulers.scheduler import BaseScheduler
def _to_tuple(x, dim=2):
......
......@@ -2,7 +2,7 @@ import math
import numpy as np
import torch
from typing import List, Optional, Tuple, Union
from lightx2v.text2v.models.schedulers.scheduler import BaseScheduler
from lightx2v.models.schedulers.scheduler import BaseScheduler
class WanScheduler(BaseScheduler):
......
import os
import torch
from lightx2v.text2v.models.video_encoders.hf.autoencoder_kl_causal_3d.autoencoder_kl_causal_3d import AutoencoderKLCausal3D
from lightx2v.text2v.models.video_encoders.trt.autoencoder_kl_causal_3d import trt_vae_infer
from lightx2v.models.video_encoders.hf.autoencoder_kl_causal_3d.autoencoder_kl_causal_3d import AutoencoderKLCausal3D
from lightx2v.models.video_encoders.trt.autoencoder_kl_causal_3d import trt_vae_infer
class VideoEncoderKLCausal3DModel:
......
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