minor fix to HF mirror script

keras-team · Oct 30, 2024 · afd6921 · afd6921
1 parent bd57aed
commit afd6921
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tools/hf_uploaded_presets.json b/tools/hf_uploaded_presets.json
@@ -145,5 +145,5 @@
     "kaggle://keras/t5/keras/flan_base_multi/2",
     "kaggle://keras/t5/keras/flan_large_multi/2",
     "kaggle://keras/xlm_roberta/keras/xlm_roberta_base_multi/2",
-    "kaggle://keras/xlm_roberta/keras/xlm_roberta_large_multi/2",
+    "kaggle://keras/xlm_roberta/keras/xlm_roberta_large_multi/2"
 ]
diff --git a/tools/mirror_weights_on_hf.py b/tools/mirror_weights_on_hf.py
@@ -87,7 +87,7 @@ def main():
 
     # Step 5: Update JSON file with newly uploaded handles
     update_hf_uploads_json(
-        JSON_FILE_PATH, {latest_kaggle_handles} - {errored_uploads}
+        JSON_FILE_PATH, list(set(latest_kaggle_handles) - set(errored_uploads))
     )
     print("uploads for the following models failed: ", errored_uploads)
     print("Rest of the models up to date on HuggingFace")