closes #113; get rid of configuring opencv

chengzeyi · Feb 5, 2024 · 307ca82 · 307ca82
1 parent 2ac39c5
commit 307ca82
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 11 deletions.
diff --git a/examples/optimize_instant_id_pipeline.py b/examples/optimize_instant_id_pipeline.py
@@ -281,9 +281,8 @@ def get_kwarg_inputs():
     # Let's see it!
     # Note: Progress bar might work incorrectly due to the async nature of CUDA.
     kwarg_inputs = get_kwarg_inputs()
-    iter_profiler = None
+    iter_profiler = IterationProfiler()
     if 'callback_on_step_end' in inspect.signature(model).parameters:
-        iter_profiler = IterationProfiler()
         kwarg_inputs[
             'callback_on_step_end'] = iter_profiler.callback_on_step_end
     begin = time.time()

diff --git a/examples/optimize_lcm_lora.py b/examples/optimize_lcm_lora.py
@@ -257,9 +257,8 @@ def get_kwarg_inputs():
     # Let's see it!
     # Note: Progress bar might work incorrectly due to the async nature of CUDA.
     kwarg_inputs = get_kwarg_inputs()
-    iter_profiler = None
+    iter_profiler = IterationProfiler()
     if 'callback_on_step_end' in inspect.signature(model).parameters:
-        iter_profiler = IterationProfiler()
         kwarg_inputs[
             'callback_on_step_end'] = iter_profiler.callback_on_step_end
     begin = time.time()

diff --git a/examples/optimize_lcm_pipeline.py b/examples/optimize_lcm_pipeline.py
@@ -257,9 +257,8 @@ def get_kwarg_inputs():
     # Let's see it!
     # Note: Progress bar might work incorrectly due to the async nature of CUDA.
     kwarg_inputs = get_kwarg_inputs()
-    iter_profiler = None
+    iter_profiler = IterationProfiler()
     if 'callback_on_step_end' in inspect.signature(model).parameters:
-        iter_profiler = IterationProfiler()
         kwarg_inputs[
             'callback_on_step_end'] = iter_profiler.callback_on_step_end
     begin = time.time()

diff --git a/examples/optimize_stable_diffusion_pipeline.py b/examples/optimize_stable_diffusion_pipeline.py
@@ -257,9 +257,8 @@ def get_kwarg_inputs():
     # Let's see it!
     # Note: Progress bar might work incorrectly due to the async nature of CUDA.
     kwarg_inputs = get_kwarg_inputs()
-    iter_profiler = None
+    iter_profiler = IterationProfiler()
     if 'callback_on_step_end' in inspect.signature(model).parameters:
-        iter_profiler = IterationProfiler()
         kwarg_inputs[
             'callback_on_step_end'] = iter_profiler.callback_on_step_end
     begin = time.time()

diff --git a/examples/optimize_stable_video_diffusion_pipeline.py b/examples/optimize_stable_video_diffusion_pipeline.py
@@ -248,9 +248,8 @@ def get_kwarg_inputs():
     # Let's see it!
     # Note: Progress bar might work incorrectly due to the async nature of CUDA.
     kwarg_inputs = get_kwarg_inputs()
-    iter_profiler = None
+    iter_profiler = IterationProfiler()
     if 'callback_on_step_end' in inspect.signature(model).parameters:
-        iter_profiler = IterationProfiler()
         kwarg_inputs[
             'callback_on_step_end'] = iter_profiler.callback_on_step_end
     begin = time.time()

diff --git a/src/sfast/utils/env.py b/src/sfast/utils/env.py
@@ -104,7 +104,7 @@ def setup_environment():
         return
     _ENV_SETUP_DONE = True
 
-    _configure_libraries()
+    # _configure_libraries()
 
     custom_module_path = os.environ.get("SFAST_ENV_MODULE")