diff --git a/src/scope/server/pipeline_artifacts.py b/src/scope/server/pipeline_artifacts.py index f2aef480..46496c39 100644 --- a/src/scope/server/pipeline_artifacts.py +++ b/src/scope/server/pipeline_artifacts.py @@ -20,12 +20,31 @@ files=["Wan2_1-VACE_module_1_3B_bf16.safetensors"], ) +# Extra VAE artifacts (lightweight/alternative encoders) +LIGHTVAE_ARTIFACT = HuggingfaceRepoArtifact( + repo_id="lightx2v/Autoencoders", + files=["lightvaew2_1.pth"], +) + +TAE_ARTIFACT = HuggingfaceRepoArtifact( + repo_id="lightx2v/Autoencoders", + files=["taew2_1.pth"], +) + +LIGHTTAE_ARTIFACT = HuggingfaceRepoArtifact( + repo_id="lightx2v/Autoencoders", + files=["lighttaew2_1.pth"], +) + # Pipeline-specific artifacts PIPELINE_ARTIFACTS = { "streamdiffusionv2": [ WAN_1_3B_ARTIFACT, UMT5_ENCODER_ARTIFACT, VACE_ARTIFACT, + LIGHTVAE_ARTIFACT, + TAE_ARTIFACT, + LIGHTTAE_ARTIFACT, HuggingfaceRepoArtifact( repo_id="jerryfeng/StreamDiffusionV2", files=["wan_causal_dmd_v2v/model.pt"], @@ -35,6 +54,9 @@ WAN_1_3B_ARTIFACT, UMT5_ENCODER_ARTIFACT, VACE_ARTIFACT, + LIGHTVAE_ARTIFACT, + TAE_ARTIFACT, + LIGHTTAE_ARTIFACT, HuggingfaceRepoArtifact( repo_id="Efficient-Large-Model/LongLive-1.3B", files=["models/longlive_base.pt", "models/lora.pt"], @@ -43,6 +65,9 @@ "krea-realtime-video": [ WAN_1_3B_ARTIFACT, UMT5_ENCODER_ARTIFACT, + LIGHTVAE_ARTIFACT, + TAE_ARTIFACT, + LIGHTTAE_ARTIFACT, HuggingfaceRepoArtifact( repo_id="Wan-AI/Wan2.1-T2V-14B", files=["config.json"], @@ -56,6 +81,9 @@ WAN_1_3B_ARTIFACT, UMT5_ENCODER_ARTIFACT, VACE_ARTIFACT, + LIGHTVAE_ARTIFACT, + TAE_ARTIFACT, + LIGHTTAE_ARTIFACT, HuggingfaceRepoArtifact( repo_id="JaydenLu666/Reward-Forcing-T2V-1.3B", files=["rewardforcing.pt"], @@ -65,6 +93,9 @@ WAN_1_3B_ARTIFACT, UMT5_ENCODER_ARTIFACT, VACE_ARTIFACT, + LIGHTVAE_ARTIFACT, + TAE_ARTIFACT, + LIGHTTAE_ARTIFACT, HuggingfaceRepoArtifact( repo_id="KlingTeam/MemFlow", files=["base.pt", "lora.pt"],