We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a769016 commit 3f2a800Copy full SHA for 3f2a800
1 file changed
src/maxdiffusion/input_pipeline/_tfds_data_processing.py
@@ -120,10 +120,10 @@ def make_tfrecord_iterator(
120
# set load_tfrecord_cached to True in config to use pre-processed tfrecord dataset.
121
# pedagogical_examples/dataset_tf_cache_to_tfrecord.py to convert tf preprocessed dataset to tfrecord.
122
# Dataset cache in github runner test doesn't contain all the features since its shared, Use the default tfrecord iterator.
123
-
+
124
# checks that the dataset path is valid. In case of gcs, the existance of the dir is not checked.
125
is_dataset_dir_valid = "gs://" in config.dataset_save_location or os.path.isdir(config.dataset_save_location)
126
127
if (
128
config.cache_latents_text_encoder_outputs
129
and is_dataset_dir_valid
0 commit comments