fix: tasksmanager path

Author: JingyaHuang

optimum/commands/export/neuron.py

@@ -18,7 +18,7 @@
 from argparse import ArgumentParser, Namespace, _SubParsersAction
 from pathlib import Path
 
-from ...exporters import TasksManager
+from ...exporters.tasks import TasksManager
 from ..base import BaseOptimumCLICommand, CommandInfo
 
 

optimum/commands/export/neuronx.py

@@ -19,7 +19,7 @@
 from argparse import SUPPRESS, ArgumentParser, Namespace, _SubParsersAction
 from pathlib import Path
 
-from ...exporters import TasksManager
+from ...exporters.tasks import TasksManager
 from ...neuron.utils.instance import SUPPORTED_INSTANCE_TYPES, normalize_instance_type
 from ...neuron.utils.system import get_neuron_major
 from ..base import BaseOptimumCLICommand, CommandInfo

optimum/neuron/cache/hub_cache.py

@@ -23,8 +23,7 @@
 from huggingface_hub import HfApi, get_token
 from huggingface_hub.errors import EntryNotFoundError
 from huggingface_hub.hf_api import RepoFile
-
-from optimum.exporters import TasksManager
+from optimum.exporters.tasks import TasksManager
 
 from ..utils.argument_utils import DTYPE_MAPPER
 from ..utils.cache_utils import get_hf_hub_cache_repo

tools/cache/auto_fill_diffusion_cache.py

@@ -22,8 +22,8 @@
 import time
 
 import requests
+from optimum.exporters.tasks import TasksManager
 
-from optimum.exporters import TasksManager
 from optimum.neuron.utils.instance import SUPPORTED_INSTANCE_TYPES
 
 

tools/cache/auto_fill_llm_cache.py

@@ -22,8 +22,8 @@
 import time
 
 import requests
+from optimum.exporters.tasks import TasksManager
 
-from optimum.exporters import TasksManager
 from optimum.neuron.utils.instance import SUPPORTED_INSTANCE_TYPES