ibid

utils/pipelines.py

@@ -133,7 +133,7 @@ def unconditional_generation_with_input_config(input_file, num_batches, num_desi
         for file_name in os.listdir(directory):
             if file_name.endswith(".cif.gz"):
                 name = os.path.basename(file_name).split(".")[0]  #filename without extension
-                terms = file_name.split("_")
+                terms = name.split("_")
                 model_index = terms.index("model")
                 batch = int(terms[model_index - 1])
                 design = int(terms[model_index + 1])