diff --git a/fooocus_version.py b/fooocus_version.py
index db872c8..bd44f54 100644
--- a/fooocus_version.py
+++ b/fooocus_version.py
@@ -1 +1 @@
-version = '2.1.677'
+version = '2.1.678'
diff --git a/modules/patch.py b/modules/patch.py
index 58f0e48..18a45ba 100644
--- a/modules/patch.py
+++ b/modules/patch.py
@@ -24,7 +24,7 @@ import warnings
 import safetensors.torch
 
 from fcbh.k_diffusion import utils
-from fcbh.k_diffusion.sampling import BrownianTreeNoiseSampler, trange
+from fcbh.k_diffusion.sampling import trange
 from fcbh.ldm.modules.diffusionmodules.openaimodel import timestep_embedding, forward_timestep_embed