Spaces:
Running
Running
add saving feature extractor
Browse files
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():
|