diff --git a/Disco_Diffusion.ipynb b/Disco_Diffusion.ipynb index aeb6be6..8054174 100644 --- a/Disco_Diffusion.ipynb +++ b/Disco_Diffusion.ipynb @@ -16,7 +16,7 @@ "id": "TitleTop" }, "source": [ - "# Disco Diffusion v5 - Now with 3D animation\n", + "# Disco Diffusion v5.1 - Now with Turbo\n", "\n", "In case of confusion, Disco is the name of this notebook edit. The diffusion model in use is Katherine Crowson's fine-tuned 512x512 model\n", "\n", @@ -58,7 +58,9 @@ "\n", "Somnai (https://twitter.com/Somnai_dreams) added Diffusion Animation techniques, QoL improvements and various implementations of tech and techniques, mostly listed in the changelog below.\n", "\n", - "3D animation implementation added by Adam Letts (https://twitter.com/gandamu_ml) in collaboration with Somnai." + "3D animation implementation added by Adam Letts (https://twitter.com/gandamu_ml) in collaboration with Somnai.\n", + "\n", + "Turbo feature by Chris Allen (https://twitter.com/zippy731)" ] }, { @@ -235,10 +237,15 @@ "\n", " Separated transform code\n", "\n", - " v5.01 Update: Match 10th 2022 - gandamu / Adam Letts\n", + " v5.01 Update: March 10th 2022 - gandamu / Adam Letts\n", "\n", " IPython magic commands replaced by Python code\n", "\n", + " v5.1 Update: March 30th 2022 - zippy / Chris Allen - c/o gandamu / Adam Letts\n", + "\n", + " Integrated Turbo+Smooth features from Disco Diffusion Turbo -- just the implementation, without its defaults.\n", + " Implemented resume of turbo animations in such a way that it's now possible to resume from different batch folders and batch numbers.\n", + "\n", " '''\n", " )\n" ], @@ -2559,8 +2566,9 @@ "#@markdown ####**Turbo Mode (3D anim only):**\n", "#@markdown (Starts after frame 10,) skips diffusion steps and just uses depth map to warp images for skipped frames.\n", "#@markdown Speeds up rendering by 2x-4x, and may improve image coherence between frames. frame_blend_mode smooths abrupt texture changes across 2 frames.\n", + "#@markdown For different settings tuned for Turbo Mode, refer to the original Disco-Turbo Github: https://github.com/zippy731/disco-diffusion-turbo\n", "\n", - "turbo_mode = True #@param {type:\"boolean\"}\n", + "turbo_mode = False #@param {type:\"boolean\"}\n", "turbo_steps = \"3\" #@param [\"2\",\"3\",\"4\",\"5\",\"6\"] {type:\"string\"}\n", "turbo_preroll = 10 # frames\n", "turbo_frame_blend = True #@param {type:\"boolean\"}\n", diff --git a/disco.py b/disco.py index 5f67d1f..231bbfe 100644 --- a/disco.py +++ b/disco.py @@ -5,7 +5,7 @@ # %% """ -# Disco Diffusion v5 - Now with 3D animation +# Disco Diffusion v5.1 - Now with Turbo In case of confusion, Disco is the name of this notebook edit. The diffusion model in use is Katherine Crowson's fine-tuned 512x512 model @@ -40,6 +40,8 @@ Disco: Somnai (https://twitter.com/Somnai_dreams) added Diffusion Animation techniques, QoL improvements and various implementations of tech and techniques, mostly listed in the changelog below. 3D animation implementation added by Adam Letts (https://twitter.com/gandamu_ml) in collaboration with Somnai. + +Turbo feature by Chris Allen (https://twitter.com/zippy731) """ # %% @@ -198,10 +200,16 @@ if skip_for_run_all == False: Separated transform code - v5.01 Update: Match 10th 2022 - gandamu / Adam Letts + v5.01 Update: March 10th 2022 - gandamu / Adam Letts IPython magic commands replaced by Python code + v5.1 Update: March 30th 2022 - zippy / Chris Allen and gandamu / Adam Letts + + Integrated Turbo+Smooth features from Disco Diffusion Turbo -- just the implementation, without its defaults. + + Implemented resume of turbo animations in such a way that it's now possible to resume from different batch folders and batch numbers. + ''' ) @@ -2415,8 +2423,9 @@ sampling_mode = 'bicubic'#@param {type:"string"} #@markdown ####**Turbo Mode (3D anim only):** #@markdown (Starts after frame 10,) skips diffusion steps and just uses depth map to warp images for skipped frames. #@markdown Speeds up rendering by 2x-4x, and may improve image coherence between frames. frame_blend_mode smooths abrupt texture changes across 2 frames. +#@markdown For different settings tuned for Turbo Mode, refer to the original Disco-Turbo Github: https://github.com/zippy731/disco-diffusion-turbo -turbo_mode = True #@param {type:"boolean"} +turbo_mode = False #@param {type:"boolean"} turbo_steps = "3" #@param ["2","3","4","5","6"] {type:"string"} turbo_preroll = 10 # frames turbo_frame_blend = True #@param {type:"boolean"}