Upload modeling_rwkv6.py

fix import error with flash linear attention

modeling_rwkv6.py

@@ -17,7 +17,6 @@
 from dataclasses import dataclass
 from typing import List, Optional, Tuple, Union
 
-from pathlib import Path
 
 import torch
 import torch.nn.functional as F
@@ -31,14 +30,12 @@ from transformers.utils import (
     add_code_sample_docstrings,
     add_start_docstrings,
     add_start_docstrings_to_model_forward,
-    is_ninja_available,
-    is_torch_cuda_available,
     logging,
 )
 
 from .configuration_rwkv6 import Rwkv6Config
 try:
-    from fla.ops.rwkv6.recurrent_fuse import fused_recurrent_rwkv6
+    from fla.ops.rwkv6 import fused_recurrent_rwkv6
 except ImportError:
     print("Required module is not installed. Please install it using the following commands:")
     print("pip install -U git+https://github.com/sustcsonglin/flash-linear-attention")