Fix dataset target: exp004 trained on diffusion-pretrain-set-ft1 data/deepfashion (has caption_vlm_json), not qwen-deepfashion-fused; shard listing filtered to the config (the repo holds 582 parquet across configs). Caught by CELL 0 refusing before spend — as designed.
Browse files- colab/anima_closeout.ipynb +12 -3
colab/anima_closeout.ipynb
CHANGED
|
@@ -138,11 +138,20 @@
|
|
| 138 |
"print(\"Anima split_files present \u2014 NC weights; derived ckpts inherit NC\")\n",
|
| 139 |
"\n",
|
| 140 |
"# -- 5. dataset columns, verified BEFORE anything spends ------------------\n",
|
| 141 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 142 |
"from huggingface_hub import HfApi\n",
|
| 143 |
"SHARDS = sorted(f for f in HfApi().list_repo_files(\n",
|
| 144 |
-
" DATASET, repo_type=\"dataset\")
|
| 145 |
-
"
|
|
|
|
|
|
|
| 146 |
"import pyarrow.parquet as pq\n",
|
| 147 |
"_p0 = hf_hub_download(DATASET, SHARDS[0], repo_type=\"dataset\")\n",
|
| 148 |
"_cols = set(pq.read_schema(_p0).names)\n",
|
|
|
|
| 138 |
"print(\"Anima split_files present \u2014 NC weights; derived ckpts inherit NC\")\n",
|
| 139 |
"\n",
|
| 140 |
"# -- 5. dataset columns, verified BEFORE anything spends ------------------\n",
|
| 141 |
+
"# exp004's actual source: diffusion-pretrain-set-ft1, deepfashion config \u2014\n",
|
| 142 |
+
"# the dataset whose shards were the pod's ft1_shards and which carries\n",
|
| 143 |
+
"# caption_vlm_json. (qwen-deepfashion-fused does NOT have that column; the\n",
|
| 144 |
+
"# first campaign run caught this exactly here, at zero cost.) The repo\n",
|
| 145 |
+
"# holds 582 parquet files across many configs \u2014 filter to deepfashion or\n",
|
| 146 |
+
"# the split cell would download all of them.\n",
|
| 147 |
+
"DATASET = \"AbstractPhil/diffusion-pretrain-set-ft1\"\n",
|
| 148 |
+
"DATASET_SUBDIR = \"data/deepfashion/\"\n",
|
| 149 |
"from huggingface_hub import HfApi\n",
|
| 150 |
"SHARDS = sorted(f for f in HfApi().list_repo_files(\n",
|
| 151 |
+
" DATASET, repo_type=\"dataset\")\n",
|
| 152 |
+
" if f.endswith(\".parquet\") and f.startswith(DATASET_SUBDIR))\n",
|
| 153 |
+
"assert SHARDS, f\"no parquet shards under {DATASET_SUBDIR} in {DATASET}\"\n",
|
| 154 |
+
"print(f\"deepfashion shards: {len(SHARDS)}\")\n",
|
| 155 |
"import pyarrow.parquet as pq\n",
|
| 156 |
"_p0 = hf_hub_download(DATASET, SHARDS[0], repo_type=\"dataset\")\n",
|
| 157 |
"_cols = set(pq.read_schema(_p0).names)\n",
|