Update clean_community_org.py
Browse files- clean_community_org.py +2 -2
clean_community_org.py
CHANGED
|
@@ -14,8 +14,8 @@ from huggingface_hub import HfApi
|
|
| 14 |
def garbage_collect_empty_models(token: str | None = None):
|
| 15 |
api = HfApi(token=token)
|
| 16 |
now = datetime.now(timezone.utc)
|
| 17 |
-
print("Running garbage collection on
|
| 18 |
-
for model in api.list_models(author="
|
| 19 |
if model.siblings and len(model.siblings) > 1:
|
| 20 |
# If model has files, then it's not empty
|
| 21 |
continue
|
|
|
|
| 14 |
def garbage_collect_empty_models(token: str | None = None):
|
| 15 |
api = HfApi(token=token)
|
| 16 |
now = datetime.now(timezone.utc)
|
| 17 |
+
print("Running garbage collection on djuna-test-lab.")
|
| 18 |
+
for model in api.list_models(author="djuna-test-lab", full=True):
|
| 19 |
if model.siblings and len(model.siblings) > 1:
|
| 20 |
# If model has files, then it's not empty
|
| 21 |
continue
|