Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit a540357c authored by helloyongyang's avatar helloyongyang
Browse files

fix ci

parent a67b9c63
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lightx2v/models/networks/wan/causvid_model.py
View file @ a540357c
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