Upload civitai_to_hf.py

civitai_to_hf.py

@@ -75,7 +75,7 @@ def get_repo_hashes(repo_id: str, repo_type: str="model"):
         if not api.repo_exists(repo_id=repo_id, repo_type=repo_type, token=hf_token): return hashes
         tree = api.list_repo_tree(repo_id=repo_id, repo_type=repo_type, token=hf_token)
         for f in tree:
-            if not isinstance(f, RepoFile) or f.lfs is None: continue
+            if not isinstance(f, RepoFile) or f.lfs is None or f.lfs.get("sha256", None) is None: continue
             hashes.append(f.lfs["sha256"])
     except Exception as e:
         print(e)