echarlaix HF staff commited on
Commit
59bef3f
1 Parent(s): 7be838c

add saving feature extractor

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ def export(
80
  print("Repo created successfully!", new_repo_url)
81
 
82
  folder = Path(folder)
83
- for dir_name in ("", "vae_encoder", "vae_decoder", "text_encoder", "text_encoder_2", "unet", "tokenizer", "tokenizer_2", "scheduler"):
84
  if not (folder / dir_name).is_dir():
85
  continue
86
  for file_path in (folder / dir_name).iterdir():
 
80
  print("Repo created successfully!", new_repo_url)
81
 
82
  folder = Path(folder)
83
+ for dir_name in ("", "vae_encoder", "vae_decoder", "text_encoder", "text_encoder_2", "unet", "tokenizer", "tokenizer_2", "scheduler", "feature_extractor"):
84
  if not (folder / dir_name).is_dir():
85
  continue
86
  for file_path in (folder / dir_name).iterdir():