diff --git a/Disco_Diffusion.ipynb b/Disco_Diffusion.ipynb index c85ca42..3da90f1 100644 --- a/Disco_Diffusion.ipynb +++ b/Disco_Diffusion.ipynb @@ -1100,9 +1100,7 @@ " skip_steps = args.calc_frames_skip_steps\n", "\n", " if args.animation_mode == \"3D\":\n", - " if frame_num == 0:\n", - " pass\n", - " else:\n", + " if frame_num > 0:\n", " seed += 1 \n", " if resume_run and frame_num == start_frame:\n", " img_filepath = batchFolder+f\"/{batch_name}({batchNum})_{start_frame-1:04}.png\"\n", @@ -1413,7 +1411,7 @@ " image.save(f'{batchFolder}/{filename}')\n", " if args.animation_mode == \"3D\":\n", " # If turbo, save a blended image\n", - " if turbo_mode:\n", + " if turbo_mode and frame_num > 0:\n", " # Mix new image with prevFrameScaled\n", " blend_factor = (1)/int(turbo_steps)\n", " newFrame = cv2.imread('prevFrame.png') # This is already updated..\n", @@ -3274,4 +3272,4 @@ }, "nbformat": 4, "nbformat_minor": 4 -} \ No newline at end of file +}