AbstractPhil commited on
Commit
8039dec
·
verified ·
1 Parent(s): 00cc430

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
Files changed (1) hide show
  1. 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
- "DATASET = \"AbstractPhil/qwen-deepfashion-fused\"\n",
 
 
 
 
 
 
 
142
  "from huggingface_hub import HfApi\n",
143
  "SHARDS = sorted(f for f in HfApi().list_repo_files(\n",
144
- " DATASET, repo_type=\"dataset\") if f.endswith(\".parquet\"))\n",
145
- "assert SHARDS, f\"no parquet shards in {DATASET}\"\n",
 
 
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",