Add tag_model function to huggingface tools

fmtr · Aug 17, 2024 · 3de4aba · 3de4aba
1 parent c79a139
commit 3de4aba
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/fmtr/tools/hfh_tools.py b/fmtr/tools/hfh_tools.py
@@ -60,5 +60,11 @@ def main():
     run(args.config)
 
 
-if __name__ == '__main__':
-    run('cache.json')
+def tag_model(repo_id: str, tag: str):
+    """
+
+    Tag a model repository
+
+    """
+    api = huggingface_hub.HfApi()
+    return api.create_tag(repo_id, tag=tag, repo_type='model')
diff --git a/fmtr/tools/version b/fmtr/tools/version
@@ -1 +1 @@
-0.8.17
+0.8.18