Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-11-04 14:16:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed Wan 2.2 VAE URL

Browse Source
This commit is contained in:
deepbeepmeep 2025-08-06 21:08:44 +02:00
parent b01a5e4bc8
commit 570cc6809f
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
models/wan/wan_handler.py
View File

@ -106,12 +106,21 @@ class family_handler():
@staticmethod
def query_model_files(computeList, base_model_type, model_filename, text_encoder_quantization):
text_encoder_filename = family_handler.get_wan_text_encoder_filename(text_encoder_quantization)
vae_filepath = "Wan2.2_VAE.safetensors" if base_model_type == "ti2v_2_2" else "Wan2.1_VAE.safetensors"
return {
download_def = [{
"repoId" : "DeepBeepMeep/Wan2.1",
"sourceFolderList" : ["xlm-roberta-large", "umt5-xxl", "" ],
"fileList" : [ [ "models_clip_open-clip-xlm-roberta-large-vit-huge-14-bf16.safetensors", "sentencepiece.bpe.model", "special_tokens_map.json", "tokenizer.json", "tokenizer_config.json"], ["special_tokens_map.json", "spiece.model", "tokenizer.json", "tokenizer_config.json"] + computeList(text_encoder_filename) , [vae_filepath, "fantasy_proj_model.safetensors" ] + computeList(model_filename) ]
}
"fileList" : [ [ "models_clip_open-clip-xlm-roberta-large-vit-huge-14-bf16.safetensors", "sentencepiece.bpe.model", "special_tokens_map.json", "tokenizer.json", "tokenizer_config.json"], ["special_tokens_map.json", "spiece.model", "tokenizer.json", "tokenizer_config.json"] + computeList(text_encoder_filename) , ["Wan2.1_VAE.safetensors", "fantasy_proj_model.safetensors" ] + computeList(model_filename) ]
}]
if base_model_type == "ti2v_2_2":
download_def += [ {
"repoId" : "DeepBeepMeep/Wan2.2",
"sourceFolderList" : [""],
"fileList" : [ [ "Wan2.2_VAE.safetensors" ] ]
}]
return download_def
@staticmethod