fix

Author: lucidrains

train_latent_with_text.py

@@ -172,7 +172,7 @@ def cycle(iter_dl):
         text_tensor = text_tensor[text_tensor < 256] # todo: offer a utility function for removing meta tags and special tokens
 
         text = decode_tokens(text_tensor)
-        filename = str(results_folder / f'{text}.{step}.png')
+        filename = str(results_folder / f'{step}.{text}.png')
 
         save_image(
             image.detach().cpu(),