Tryng to train on custome dataset but not working "--caption_column` value 'text' not found"

nub29 · July 1, 2023, 2:18pm

huggingface/diffusers/blob/main/examples/controlnet/train_controlnet.py

#!/usr/bin/env python
# coding=utf-8
# Copyright 2023 The HuggingFace Inc. team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and

import argparse
import logging
import math
import os
import random

This file has been truncated. show original

i try formatting the the dataset to how the example was downloading
{“text”: “pale golden rod circle with old lace background”, “image”: “images/0.png”, “conditioning_image”: “conditioning_images/0.png”}
putting the images in the same directory iun that format but it simpyl does not work

I but keep getting this error

ValueError: --caption_column value ‘text’ not found in dataset columns. Dataset columns are: image, label

im not sure what else i am suppose to do?

mariosasko · July 2, 2023, 3:12pm

You need to create a dataset script to train on a custom dataset, but based on the error, you seem to be using the imagefolder loader (doesn’t support datasets with more than one image column) to build the dataset. Let us know if you need help writing the script.

abovzv · April 23, 2024, 12:49am

I have encountered the same problem as described here, so how should I solve this problem?

Topic		Replies	Views
New Fine-tuner Question/Struggles 🧨 Diffusers	6	3336	February 1, 2023
Text Column not working with Image Folder 🤗Datasets	6	633	August 29, 2023
Data format for text-to-image 🧨 Diffusers	3	1692	January 14, 2023
Couldn't find 'my_dataset' on the Hugging Face Hub 🤗Datasets	4	2720	May 2, 2023
ValueError: Need either a dataset name or a training/validation file Beginners	1	666	June 24, 2023

Tryng to train on custome dataset but not working "--caption_column` value 'text' not found"

Related topics