step

huggingface · Jan 27, 2025 · fa7e2e8 · fa7e2e8
1 parent 147c30e
commit fa7e2e8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/optimum/exporters/neuron/__main__.py b/optimum/exporters/neuron/__main__.py
@@ -665,9 +665,6 @@ def main_export(
         ip_adapter_scales=ip_adapter_scales,
         **input_shapes,
     )
-    import pdb
-
-    pdb.set_trace()
 
     _, neuron_outputs = export_models(
         models_and_neuron_configs=models_and_neuron_configs,

diff --git a/optimum/exporters/neuron/convert.py b/optimum/exporters/neuron/convert.py
@@ -351,6 +351,8 @@ def export_models(
     failed_models = []
     total_compilation_time = 0
     compile_configs = {}
+    import pdb
+    pdb.set_trace()
     for i, model_name in enumerate(models_and_neuron_configs.keys()):
         logger.info(f"***** Compiling {model_name} *****")
         submodel, sub_neuron_config = models_and_neuron_configs[model_name]

diff --git a/optimum/neuron/utils/__init__.py b/optimum/neuron/utils/__init__.py
@@ -48,6 +48,7 @@
         "DummyMaskedPosGenerator",
         "DummyControNetInputGenerator",
         "ASTDummyAudioInputGenerator",
+        "DummyIPAdapterInputGenerator",
     ],
     "misc": [
         "DiffusersPretrainedConfig",
@@ -110,6 +111,7 @@
         DummyBeamValuesGenerator,
         DummyControNetInputGenerator,
         DummyMaskedPosGenerator,
+        DummyIPAdapterInputGenerator,
     )
     from .misc import (
         DiffusersPretrainedConfig,