diff --git a/optimum/exporters/neuron/__main__.py b/optimum/exporters/neuron/__main__.py index d31f6f53b..004c37c7a 100644 --- 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 index b9b1a79c5..f1f941365 100644 --- 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 index 5f8b7fca0..c6bac350e 100644 --- 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,