DeepBeepMeep commited on
Commit
2b5307e
·
1 Parent(s): 50a1f67
Files changed (1) hide show
  1. wgp.py +2 -2
wgp.py CHANGED
@@ -1930,7 +1930,7 @@ def download_models(transformer_filename):
1930
  model_def = {
1931
  "repoId" : "DeepBeepMeep/HunyuanVideo",
1932
  "sourceFolderList" : [ "llava-llama-3-8b", "clip_vit_large_patch14", "" ],
1933
- "fileList" :[ ["config.json", "special_tokens_map.json", "tokenizer.json", "tokenizer_config.json", "preprocessor_config.json"] + computeList(text_encoder_filename) , ["config.json", "model.safetensors", "preprocessor_config.json", "special_tokens_map.json", "tokenizer.json", "tokenizer_config.json", "vocab.json"], [ "hunyuan_video_720_quanto_int8_map.json", "hunyuan_video_custom_VAE_fp32.safetensors", "hunyuan_video_custom_VAE_config.json", "hunyuan_video_VAE_fp32.safetensors", "hunyuan_video_VAE_config.json" , "hunyuan_video_720_quanto_int8_map.json" ] + computeList(transformer_filename) ]
1934
  }
1935
 
1936
  else:
@@ -2307,7 +2307,7 @@ def apply_changes( state,
2307
  reload_needed = True
2308
  model_choice = generate_dropdown_model_list()
2309
 
2310
- header = generate_header(transformer_filename, compile=compile, attention_mode= attention_mode)
2311
  return "<DIV ALIGN=CENTER>The new configuration has been succesfully applied</DIV>", header, model_choice, gr.update(visible= server_config["enhancer_enabled"] == 1)
2312
 
2313
 
 
1930
  model_def = {
1931
  "repoId" : "DeepBeepMeep/HunyuanVideo",
1932
  "sourceFolderList" : [ "llava-llama-3-8b", "clip_vit_large_patch14", "" ],
1933
+ "fileList" :[ ["config.json", "special_tokens_map.json", "tokenizer.json", "tokenizer_config.json", "preprocessor_config.json"] + computeList(text_encoder_filename) , ["config.json", "merges.txt", "model.safetensors", "preprocessor_config.json", "special_tokens_map.json", "tokenizer.json", "tokenizer_config.json", "vocab.json"], [ "hunyuan_video_720_quanto_int8_map.json", "hunyuan_video_custom_VAE_fp32.safetensors", "hunyuan_video_custom_VAE_config.json", "hunyuan_video_VAE_fp32.safetensors", "hunyuan_video_VAE_config.json" , "hunyuan_video_720_quanto_int8_map.json" ] + computeList(transformer_filename) ]
1934
  }
1935
 
1936
  else:
 
2307
  reload_needed = True
2308
  model_choice = generate_dropdown_model_list()
2309
 
2310
+ header = generate_header(state["model_filename"], compile=compile, attention_mode= attention_mode)
2311
  return "<DIV ALIGN=CENTER>The new configuration has been succesfully applied</DIV>", header, model_choice, gr.update(visible= server_config["enhancer_enabled"] == 1)
2312
 
2313