diff --git a/src/transformers/models/depth_pro/configuration_depth_pro.py b/src/transformers/models/depth_pro/configuration_depth_pro.py index a93c65a79969a9..7f0b54ef4f4b00 100644 --- a/src/transformers/models/depth_pro/configuration_depth_pro.py +++ b/src/transformers/models/depth_pro/configuration_depth_pro.py @@ -168,3 +168,5 @@ def __init__( self.scaled_images_ratios = scaled_images_ratios self.scaled_images_overlap_ratios = scaled_images_overlap_ratios self.scaled_images_feature_dims = scaled_images_feature_dims + +__all__ = ["DepthProConfig"] diff --git a/src/transformers/models/depth_pro/image_processing_depth_pro.py b/src/transformers/models/depth_pro/image_processing_depth_pro.py index fb31e9084abe80..24104c0d5cbc14 100644 --- a/src/transformers/models/depth_pro/image_processing_depth_pro.py +++ b/src/transformers/models/depth_pro/image_processing_depth_pro.py @@ -407,3 +407,5 @@ def post_process_depth_estimation( outputs["predicted_depth"].append(predicted_depth) return outputs + +__all__ = ["DepthProImageProcessor"] diff --git a/src/transformers/models/depth_pro/image_processing_depth_pro_fast.py b/src/transformers/models/depth_pro/image_processing_depth_pro_fast.py index 12e56d69dfffd5..55f987aa48b766 100644 --- a/src/transformers/models/depth_pro/image_processing_depth_pro_fast.py +++ b/src/transformers/models/depth_pro/image_processing_depth_pro_fast.py @@ -380,3 +380,5 @@ def post_process_depth_estimation( outputs["predicted_depth"].append(predicted_depth) return outputs + +__all__ = ["DepthProImageProcessorFast"] diff --git a/src/transformers/models/depth_pro/modeling_depth_pro.py b/src/transformers/models/depth_pro/modeling_depth_pro.py index bc946dceb3143d..44ace2808039ec 100644 --- a/src/transformers/models/depth_pro/modeling_depth_pro.py +++ b/src/transformers/models/depth_pro/modeling_depth_pro.py @@ -1674,3 +1674,5 @@ def forward( hidden_states=depth_pro_outputs.hidden_states, attentions=depth_pro_outputs.attentions, ) + +__all__ = ["DepthProPreTrainedModel", "DepthProModel", "DepthProForDepthEstimation"]