debug

app.py

@@ -192,6 +192,49 @@ except ImportError as e:
             torchaudio.save(output_path, waveform, sr)
         return output_path
 
+# 修复可能存在的递归调用问题
+# 检查是否在运行时发生了transformers库中的注意力机制递归
+try:
+    import transformers
+    from transformers.models.llama.modeling_llama import LlamaAttention, LlamaModel
+    
+    # 保存原始的注意力前向函数
+    if hasattr(LlamaAttention, "forward"):
+        original_attention_forward = LlamaAttention.forward
+        
+        # 创建防止递归的补丁函数
+        def safe_attention_forward(self, *args, **kwargs):
+            # 使用原始函数，但避免递归调用
+            return original_attention_forward(self, *args, **kwargs)
+        
+        # 替换原始函数
+        LlamaAttention.forward = safe_attention_forward
+        print("已修复LlamaAttention.forward，防止递归")
+    
+    # 可能存在其他递归路径
+    if hasattr(transformers.models.llama.modeling_llama, "LlamaAttention"):
+        for attr_name in dir(transformers.models.llama.modeling_llama.LlamaAttention):
+            if attr_name.startswith("_") and "forward" in attr_name:
+                attr = getattr(transformers.models.llama.modeling_llama.LlamaAttention, attr_name)
+                if callable(attr):
+                    # 保存原始函数
+                    setattr(transformers.models.llama.modeling_llama.LlamaAttention, 
+                            f"original_{attr_name}", attr)
+                    
+                    # 创建安全函数
+                    def create_safe_function(original_func, attr_name):
+                        def safe_function(self, *args, **kwargs):
+                            return original_func(self, *args, **kwargs)
+                        return safe_function
+                    
+                    # 替换函数
+                    setattr(transformers.models.llama.modeling_llama.LlamaAttention, 
+                            attr_name, 
+                            create_safe_function(attr, attr_name))
+                    print(f"已修复潜在的递归函数: {attr_name}")
+except Exception as e:
+    print(f"应用注意力机制补丁时出错: {str(e)}")
+
 # 模型配置常量
 REPO_ID = "amphion/Vevo"
 CACHE_DIR = "./ckpts/Vevo"