diff --git a/backends/neuron/tests/prune_test_models.py b/backends/neuron/tests/prune_test_models.py new file mode 100644 index 00000000..592b1070 --- /dev/null +++ b/backends/neuron/tests/prune_test_models.py @@ -0,0 +1,25 @@ +import os + +from argparse import ArgumentParser +from huggingface_hub import HfApi + + +def main(): + parser = ArgumentParser() + parser.add_argument("--yes", action="store_true", default=False) + args = parser.parse_args() + api = HfApi() + models = api.list_models(search="optimum-internal-testing/neuron-tgi-testing") + for model in models: + if args.yes: + delete = True + else: + answer = input(f"Do you want to delete {model.id} [y/N] ?") + delete = (answer == "y") + if delete: + api.delete_repo(model.id) + print(f"Deleted {model.id}.") + + +if __name__ == "__main__": + main()