Skip to content

Commit 92280ef

Browse files
authored
Merge pull request #43 from xianbaoqian/xianbao-update
update local ds code snippet
2 parents bd43318 + 561a747 commit 92280ef

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

hackathon/dreambooth.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -370,6 +370,7 @@
370370
]
371371
},
372372
{
373+
"attachments": {},
373374
"cell_type": "markdown",
374375
"id": "c1ef3d21-6e0e-46c9-a459-8a2ab856a5ca",
375376
"metadata": {},
@@ -380,10 +381,9 @@
380381
"from datasets import load_dataset\n",
381382
"\n",
382383
"dataset = load_dataset(\"imagefolder\", data_dir=\"your_folder_of_images\")\n",
383-
"# Remove the dummy label column\n",
384-
"dataset = dataset.remove_columns(\"label\")\n",
385384
"# Push to Hub\n",
386385
"dataset.push_to_hub(\"dreambooth-hackathon-images\")\n",
386+
"dataset = dataset['train']\n",
387387
"```"
388388
]
389389
},

0 commit comments

Comments
 (0)