Update app.py

app.py

@@ -28,59 +28,6 @@ if not os.path.exists(REPO_PATH):
         print(f"Error cloning repository: {e.stderr.decode()}")
         sys.exit(1)
 
-attention_file_path = os.path.join(REPO_PATH, "longcat_video", "modules", "attention.py")
-try:
-    with open(attention_file_path, "r") as f:
-        content = f.read()
-
-    # Original code block that we need to replace
-    original_code = """            x, *_ = flash_attn_func(
-                q,
-                k,
-                v,
-                softmax_scale=self.scale,
-            )
-            x = rearrange(x, "B S H D -> B H S D")"""
-
-    # Corrected code block to handle FA3's 3D output shape
-    corrected_code = """            x, *_ = flash_attn_func(
-                q,
-                k,
-                v,
-                softmax_scale=self.scale,
-            )
-            # The output of FA3's flash_attn_func can be 3D (total_tokens, H, D).
-            # We need to robustly reshape it back to the 4D format (B, S, H, D) that the
-            # subsequent rearrange operation expects.
-            if x.ndim == 3:
-                # B is the original batch size from the input q tensor
-                B = q.shape[0]
-                # S_total is the flattened batch and sequence length
-                S_total, H, D = x.shape
-                # Calculate the sequence length per batch item
-                S = S_total // B
-                x = x.view(B, S, H, D)
-
-            x = rearrange(x, "B S H D -> B H S D")"""
-
-    if original_code in content:
-        print("="*50)
-        print("Applying file patch to attention.py for FlashAttention-3 compatibility.")
-        content = content.replace(original_code, corrected_code)
-        with open(attention_file_path, "w") as f:
-            f.write(content)
-        print("Patch applied successfully.")
-        print("="*50)
-    else:
-        print("Attention.py already seems to be patched or has changed. Skipping patch.")
-
-except FileNotFoundError:
-    print(f"Error: Could not find {attention_file_path} to patch.")
-    sys.exit(1)
-
-# Add the cloned repository to the Python path to allow imports
-sys.path.insert(0, os.path.abspath(REPO_PATH))
-
 # Now that the repo is in the path, we can import its modules
 from huggingface_hub import snapshot_download
 from longcat_video.pipeline_longcat_video import LongCatVideoPipeline