Update generate.py

generate.py

@@ -23,7 +23,7 @@ def generate_sequence(sequence_length: int, tokenizer, mdlm: Diffusion):
 
 
 @hydra.main(version_base=None, config_path='configs', config_name='config')
-def mdlm_motif_benchmark(config):
+def generate_de_novo(config):
     path = "/workspace/sg666/MDpLM"
 
     tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t30_150M_UR50D")
@@ -57,4 +57,4 @@ def mdlm_motif_benchmark(config):
 
     
 if __name__ == "__main__":
-    mdlm_motif_benchmark()
+    generate_de_novo()