As the title, I found here is weird. So I'd like to report it as a bug.
-
the data class in MJHQ.py named "DCI" at line 49
class DCI(datasets.GeneratorBasedBuilder):
-
in func _info, the feature is
{
"filename": datasets.Value("string"),
"category": datasets.Value("string"),
"image": datasets.Image(),
"prompt": datasets.Value("string"),
"prompt_path": datasets.Value("string"),
"image_root": datasets.Value("string"),
"image_path": datasets.Value("string"),
"split": datasets.Value("string"),
}
but, in func _generate_examples, it is
{
"filename": name,
"category": category,
"image": Image.open(image_path) if self.config.return_gt else None,
"prompt": prompt,
"meta_path": meta_path,
"image_root": image_root,
"image_path": image_path,
"split": self.config.name,
}
here is the difference at the 5th feature("prompt_path" and "meta_path")