"tests/git@developer.sourcefind.cn:ci-demos/wan2.1.git" did not exist on "75f4505072f76a65f459bec4bf74466f2e5bf114"
Commit a540357c authored by helloyongyang's avatar helloyongyang
Browse files

fix ci

parent a67b9c63
import os import os
import torch import torch
from lightx2v.utils.utils import find_torch_model_path
from lightx2v.common.ops.attn.radial_attn import MaskMap from lightx2v.common.ops.attn.radial_attn import MaskMap
from lightx2v.models.networks.wan.infer.causvid.transformer_infer import ( from lightx2v.models.networks.wan.infer.causvid.transformer_infer import (
...@@ -16,6 +15,7 @@ from lightx2v.models.networks.wan.weights.transformer_weights import ( ...@@ -16,6 +15,7 @@ from lightx2v.models.networks.wan.weights.transformer_weights import (
WanTransformerWeights, WanTransformerWeights,
) )
from lightx2v.utils.envs import * from lightx2v.utils.envs import *
from lightx2v.utils.utils import find_torch_model_path
class WanCausVidModel(WanModel): class WanCausVidModel(WanModel):
......
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