space before equal sign

lucidrains · Jun 15, 2021 · 7c631fb · 7c631fb
1 parent dda155f
commit 7c631fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_dalle.py b/train_dalle.py
@@ -178,7 +178,7 @@ def cp_path_to_dir(cp_path, tag):
         assert dalle_path.exists(), 'DALL-E model file does not exist'
     loaded_obj = torch.load(str(dalle_path), map_location='cpu')
 
-    dalle_params, vae_params, weights= loaded_obj['hparams'], loaded_obj['vae_params'], loaded_obj['weights']
+    dalle_params, vae_params, weights = loaded_obj['hparams'], loaded_obj['vae_params'], loaded_obj['weights']
     opt_state = loaded_obj.get('opt_state')
     scheduler_state = loaded_obj.get('scheduler_state')