diff --git a/invokeai/backend/__init__.py b/invokeai/backend/__init__.py
index 06066dd6b1..662700e9ba 100644
--- a/invokeai/backend/__init__.py
+++ b/invokeai/backend/__init__.py
@@ -10,6 +10,7 @@ from .generator import (
     Img2Img,
     Inpaint
 )
+from .stable_diffusion.diffusers_pipeline import PipelineIntermediateState
 from .model_management import ModelManager
 from .safety_checker import SafetyChecker
 from .args import Args
diff --git a/invokeai/backend/generator/base.py b/invokeai/backend/generator/base.py
index 125f208b0a..e30b77ec33 100644
--- a/invokeai/backend/generator/base.py
+++ b/invokeai/backend/generator/base.py
@@ -154,6 +154,7 @@ class InvokeAIGenerator(metaclass=ABCMeta):
         for i in iteration_count:
             results = generator.generate(prompt,
                                          conditioning=(uc, c, extra_conditioning_info),
+                                         step_callback=step_callback,
                                          sampler=scheduler,
                                          **generator_args,
                                          )