Update __init__.py

__init__.py

@@ -1,6 +1,7 @@
-from .custom_generate import sep_cache_utils
-from .custom_generate import monkey_patching_utils
-from .custom_generate import functions_2_patch
+from . import sep_cache_utils
+from . import monkey_patching_utils
+from . import functions_2_patch
+import custom_generate
 
 
 __all__ = ["custom_generate", "sep_cache_utils", "monkey_patching_utils", "functions_2_patch"]