From 8d90e3ccf08c8103a36c30dc6ca702cd5faa3c8d Mon Sep 17 00:00:00 2001 From: narugo1992 Date: Wed, 15 May 2024 01:15:13 +0800 Subject: [PATCH] dev(narugo): code save, ci skip --- zoo/wd14/sync.py | 20 ++++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) diff --git a/zoo/wd14/sync.py b/zoo/wd14/sync.py index c63c40c7b8..7824bd02ac 100644 --- a/zoo/wd14/sync.py +++ b/zoo/wd14/sync.py @@ -120,23 +120,27 @@ def _is_fc(name): assert tags_data.shape == (1, _get_model_tags_length(model_name)) assert embeddings.shape == (1, emb_width) + if hf_fs.exists(f'datasets/deepghs/wd14_tagger_inversion/{model_name}/samples_200.npz'): + _make_inverse( + model_name=model_name, + dst_dir=os.path.join(td, MODEL_NAMES[model_name]), + onnx_model_file=onnx_file, + scale=2000, + ) + invertible = True + else: + invertible = False + records.append({ 'Name': model_name, 'Source Repository': f'[{MODEL_NAMES[model_name]}](https://huggingface.co/{MODEL_NAMES[model_name]})', 'Tags Count': _get_model_tags_length(model_name), 'Embedding Width': emb_width, + 'Inverse Supported': 'Yes' if invertible else 'No', }) _get_model_file.cache_clear() _get_model_tags_length.cache_clear() - if hf_fs.exists(f'datasets/deepghs/wd14_tagger_inversion/{model_name}/samples_200.npz'): - _make_inverse( - model_name=model_name, - dst_dir=os.path.join(td, MODEL_NAMES[model_name]), - onnx_model_file=onnx_file, - scale=2000, - ) - df_records = pd.DataFrame(records) with open(os.path.join(td, 'README.md'), 'w') as f: print('---', file=f)