chore: remove un-needed cp imports (#532)

Author: DefTruth

src/cache_dit/__init__.py

@@ -72,43 +72,6 @@ def disable_compute_comm_overlap():
         pass
 
 
-try:
-    from cache_dit.parallelism import disable_ulysses_anything
-    from cache_dit.parallelism import enable_ulysses_anything
-    from cache_dit.parallelism import disable_ulysses_anything_float8
-    from cache_dit.parallelism import enable_ulysses_anything_float8
-    from cache_dit.parallelism import disable_ulysses_float8
-    from cache_dit.parallelism import enable_ulysses_float8
-
-except ImportError as e:  # noqa: F841
-    err_msg = str(e)
-
-    def _raise_import_error(func_name: str):  # noqa: F811
-        raise ImportError(
-            f"{func_name} requires additional dependencies. "
-            "Please install cache-dit[parallelism] or cache-dit[all] "
-            f"to use this feature. Error message: {err_msg}"
-        )
-
-    def enable_ulysses_anything(*args, **kwargs):
-        _raise_import_error("enable_ulysses_anything")
-
-    def disable_ulysses_anything(*args, **kwargs):
-        _raise_import_error("disable_ulysses_anything")
-
-    def enable_ulysses_anything_float8(*args, **kwargs):
-        _raise_import_error("enable_ulysses_anything_float8")
-
-    def disable_ulysses_anything_float8(*args, **kwargs):
-        _raise_import_error("disable_ulysses_anything_float8")
-
-    def enable_ulysses_float8(*args, **kwargs):
-        _raise_import_error("enable_ulysses_float8")
-
-    def disable_ulysses_float8(*args, **kwargs):
-        _raise_import_error("disable_ulysses_float8")
-
-
 NONE = CacheType.NONE
 DBCache = CacheType.DBCache
 DBPrune = CacheType.DBPrune

src/cache_dit/caching/cache_blocks/pattern_base.py

@@ -23,8 +23,8 @@
 except ImportError:
     ContextParallelSplitHook = None
     logger.warning(
-        "Context parallelism requires the 'diffusers>=0.36.dev0'."
-        "Please install latest version of diffusers from source: \n"
+        "Context parallelism in cache-dit requires 'diffusers>=0.36.dev0.\n"
+        "Please install latest version of diffusers from source via: \n"
         "pip3 install git+https://github.com/huggingface/diffusers.git"
     )
 

src/cache_dit/parallelism/__init__.py

@@ -1,10 +1,4 @@
 from cache_dit.parallelism.parallel_backend import ParallelismBackend
 from cache_dit.parallelism.parallel_config import ParallelismConfig
-from cache_dit.parallelism.backends.native_diffusers import enable_ulysses_anything
-from cache_dit.parallelism.backends.native_diffusers import disable_ulysses_anything
-from cache_dit.parallelism.backends.native_diffusers import enable_ulysses_anything_float8
-from cache_dit.parallelism.backends.native_diffusers import disable_ulysses_anything_float8
-from cache_dit.parallelism.backends.native_diffusers import enable_ulysses_float8
-from cache_dit.parallelism.backends.native_diffusers import disable_ulysses_float8
 from cache_dit.parallelism.parallel_interface import enable_parallelism
 from cache_dit.parallelism.parallel_interface import maybe_pad_prompt

src/cache_dit/parallelism/backends/native_diffusers/context_parallelism/attention/_templated_ulysses.py

@@ -15,7 +15,6 @@
         "Please install latest version of diffusers from source: \n"
         "pip3 install git+https://github.com/huggingface/diffusers.git"
     )
-from cache_dit.logger import init_logger
 from ._distributed_primitives import (
     _get_rank_world_size,
     _gather_size_by_comm,
@@ -27,8 +26,11 @@
     _all_to_all_single,
 )
 
+from cache_dit.logger import init_logger
+
 logger = init_logger(__name__)
 
+
 __all__ = [
     "TemplatedUlyssesAnythingAttention",
     "TemplatedUlyssesAnythingAttentionFloat8",