fix: command paths

Author: JingyaHuang

optimum/commands/env.py

@@ -23,7 +23,7 @@
 from ..neuron.version import __sdk_version__ as neuron_sdk_version
 from ..neuron.version import __version__ as optimum_neuron_version
 from ..version import __version__ as optimum_version
-from . import BaseOptimumCLICommand, CommandInfo
+from .base import BaseOptimumCLICommand, CommandInfo
 
 
 class EnvironmentCommand(BaseOptimumCLICommand):

optimum/commands/register/register_export.py

@@ -15,7 +15,7 @@
 """Registers the export command for Neuron to the Optimum CLI."""
 
 from ...neuron.utils import is_neuron_available, is_neuronx_available
-from ..export import ExportCommand
+from ..export.base import ExportCommand
 
 
 _neuron_export_command_was_imported = False

optimum/exporters/neuron/utils.py

@@ -38,24 +38,16 @@
     neuron_scaled_dot_product_attention,
 )
 from ...utils import (
-    DIFFUSERS_MINIMUM_VERSION,
-    check_if_diffusers_greater,
     is_diffusers_available,
     logging,
 )
-from ...utils.import_utils import _diffusers_version
 from ..tasks import TasksManager
 
 
 logger = logging.get_logger()
 
 
 if is_diffusers_available():
-    if not check_if_diffusers_greater(DIFFUSERS_MINIMUM_VERSION.base_version):
-        raise ImportError(
-            f"We found an older version of diffusers {_diffusers_version} but we require diffusers to be >= {DIFFUSERS_MINIMUM_VERSION}. "
-            "Please update diffusers by running `pip install --upgrade diffusers`"
-        )
     from diffusers import (
         ControlNetModel,
         DiffusionPipeline,