diff --git a/caikit_computer_vision/modules/text_to_image/__init__.py b/caikit_computer_vision/modules/text_to_image/__init__.py index 1c5180f..dcc5cfa 100644 --- a/caikit_computer_vision/modules/text_to_image/__init__.py +++ b/caikit_computer_vision/modules/text_to_image/__init__.py @@ -13,4 +13,4 @@ # limitations under the License. # Local -from .sdxl import SDXLStub +from .sdxl_stub import SDXLStub diff --git a/caikit_computer_vision/modules/text_to_image/sdxl.py b/caikit_computer_vision/modules/text_to_image/sdxl_stub.py similarity index 100% rename from caikit_computer_vision/modules/text_to_image/sdxl.py rename to caikit_computer_vision/modules/text_to_image/sdxl_stub.py