diff --git a/examples/pipelines/controlnet-interior-design/pipeline.py b/examples/pipelines/controlnet-interior-design/pipeline.py index f44f07d24..be27ecb7e 100644 --- a/examples/pipelines/controlnet-interior-design/pipeline.py +++ b/examples/pipelines/controlnet-interior-design/pipeline.py @@ -47,7 +47,6 @@ "max_new_tokens": 50, }, number_of_gpus=1, - node_pool_name="model-inference-pool", ) segment_images_op = ComponentOp.from_registry( name="segment_images", @@ -56,7 +55,6 @@ "batch_size": 2, }, number_of_gpus=1, - node_pool_name="model-inference-pool", ) write_to_hub_controlnet = ComponentOp( diff --git a/examples/pipelines/finetune_stable_diffusion/pipeline.py b/examples/pipelines/finetune_stable_diffusion/pipeline.py index 5349cc7c6..bf8e1209c 100644 --- a/examples/pipelines/finetune_stable_diffusion/pipeline.py +++ b/examples/pipelines/finetune_stable_diffusion/pipeline.py @@ -68,7 +68,6 @@ "max_new_tokens": 50, }, number_of_gpus=1, - node_pool_name="model-inference-pool", ) write_to_hub = ComponentOp( @@ -80,7 +79,6 @@ "image_column_names": ["images_data"], }, number_of_gpus=1, - node_pool_name="model-inference-pool", ) pipeline = Pipeline(pipeline_name=pipeline_name, base_path=PipelineConfigs.BASE_PATH)