mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-20 06:12:07 +00:00
24 lines
647 B
Python
24 lines
647 B
Python
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()
|