Update generate.py

generate.py

@@ -355,7 +355,8 @@ class Generator(nn.Module):
             clean_latent = self.ddim_sample(self.init_noise, conds)
             torch.cuda.empty_cache()
             clean_frames = self.decode_latents_batch(clean_latent)
-            cur_output_path = os.path.join(output_path, edit_name)
+            # cur_output_path = os.path.join(output_path, edit_name)
+            cur_output_path = os.path.join("./", edit_name)
             save_config(self.config, cur_output_path, gene = True)
             save_video(clean_frames, cur_output_path, save_frame = self.save_frame)