"src/targets/gpu/device/acosh.cpp" did not exist on "6d360afdb042b83c6dd23988527644f128c71518"
Commit 75c03057 authored by helloyongyang's avatar helloyongyang
Browse files

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

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