Update custom_generate/generate.py

custom_generate/generate.py

@@ -1,3 +1,33 @@
+def debug_imports():
+    import sys
+    import os
+    import inspect
+    
+    print("\n===== 导入调试信息 =====")
+    print(f"当前工作目录: {os.getcwd()}")
+    print(f"脚本路径: {os.path.abspath(__file__)}")
+    print(f"脚本所在目录: {os.path.dirname(os.path.abspath(__file__))}")
+    print(f"父目录: {os.path.dirname(os.path.dirname(os.path.abspath(__file__)))}")
+    print(f"Python路径(sys.path):")
+    for p in sys.path:
+        print(f"  - {p}")
+    
+    try:
+        from .. import monkey_patching_utils
+        print("成功导入 monkey_patching_utils")
+    except ImportError as e:
+        print(f"导入失败: {e}")
+        print(f"当前包: {__package__}")
+        frame = inspect.currentframe().f_back
+        print(f"调用栈: {inspect.getframeinfo(frame)}")
+    
+    print("=======================\n")
+
+# 在脚本开头调用
+debug_imports()
+
+
+
 import torch
 
 import types
@@ -8,6 +38,7 @@ import torch.nn as nn
 from transformers.modeling_utils import  PreTrainedModel
 
 
+
 from ..functions_2_patch import _validate_model_kwargs, llama_atten_forward
 from ..monkey_patching_utils import monkey_patching
 from ..sep_cache_utils import SepCache