Update README.md

README.md

@@ -33,7 +33,6 @@ import argparse
 parser = argparse.ArgumentParser()
 parser.add_argument('--temperature', type=float, default=0.5)
 parser.add_argument('--repetition_penalty', type=float, default=1.1)
-parser.add_argument('--split_num', type=int, default=0)
 args = parser.parse_args()
 args.bf16 = True
 args.stream_chat = False
@@ -48,7 +47,7 @@ if not os.path.exists(predictions_dir):
     except:
         pass
 
-filename_list = [filename for filename in os.listdir(test_data_dir) if filename.endswith(".png") and int(filename[:-4]) % 4 == args.split_num]
+filename_list = [filename for filename in os.listdir(test_data_dir) if filename.endswith(".png")]
 world_size = 1
 model, model_args = FineTuneTestCogAgentModel.from_pretrained(
         f"/path/to/design2code-18b-v0",