Update app.py

app.py

@@ -1500,83 +1500,141 @@ def download_gguf_set(
 
     repo_id = clean_repo_id(repo_id)
 
-    paths = get_bucket_paths(repo_id)
+    project = get_project_name(repo_id)
+
+    cache_dir = os.path.join(
+        BUCKET_DIR,
+        "gguf",
+        project
+    )
+
+    os.makedirs(
+        cache_dir,
+        exist_ok=True
+    )
+
+    log(
+        f"🪣 BUCKET CACHE:\n{cache_dir}"
+    )
 
-    bf16_dir = paths["bf16"]
 
     files = api.list_repo_files(
         repo_id,
         repo_type="model"
     )
 
-    shards = sorted([
-        f
-        for f in files
-        if (
-            f.lower().endswith(".gguf")
-            and "bf16" in f.lower()
-        )
-    ])
+
+    shards = sorted(
+        [
+            f for f in files
+            if (
+                f.lower().endswith(".gguf")
+                and "bf16" in f.lower()
+            )
+        ]
+    )
+
 
     if not shards:
+
         raise RuntimeError(
             "No BF16 GGUF shards found"
         )
 
+
     log(
-        f"📦 Found {len(shards)} BF16 shards"
+        f"📦 Found {len(shards)} shards"
     )
 
+
     local_files = []
 
-    for idx, shard in enumerate(
+
+    for idx, remote in enumerate(
         shards,
         start=1
     ):
 
-        local_path = os.path.join(
-            bf16_dir,
-            os.path.basename(shard)
+        filename = os.path.basename(
+            remote
+        )
+
+        local = os.path.join(
+            cache_dir,
+            filename
+        )
+
+
+        log(
+            f"📥 SHARD {idx}/{len(shards)}"
+        )
+
+        log(
+            remote
         )
 
-        if os.path.exists(local_path):
+
+        if os.path.exists(local):
+
+            size = (
+                os.path.getsize(local)
+                /
+                (1024**3)
+            )
 
             log(
-                f"♻️ SHARD {idx} already cached"
+                f"♻️ Already exists {size:.2f} GB"
             )
 
         else:
 
-            log(
-                f"📥 Downloading shard {idx}/{len(shards)}"
-            )
+            downloaded = hf_hub_download(
 
-            hf_hub_download(
                 repo_id=repo_id,
-                filename=shard,
-                local_dir=bf16_dir,
+
+                filename=remote,
+
                 repo_type="model",
-                token=HF_TOKEN
+
+                local_dir=cache_dir,
+
+                token=HF_TOKEN,
+
+            )
+
+
+            if downloaded != local:
+
+                shutil.move(
+                    downloaded,
+                    local
+                )
+
+
+            size = (
+                os.path.getsize(local)
+                /
+                (1024**3)
             )
 
             log(
-                f"✅ Saved {os.path.basename(shard)}"
+                f"✅ Saved {size:.2f} GB"
             )
 
-        show_bucket()
 
         local_files.append(
-            local_path
+            local
         )
 
-    create_bf16_flag(
-        repo_id,
-        [
-            os.path.basename(x)
-            for x in local_files
-        ]
+
+        show_bucket()
+
+
+    log(
+        "🎉 ALL BF16 SHARDS READY"
     )
 
+
     return local_files[0]
 
 # =====================================================