qv

Goekdeniz-Guelmez · Jul 31, 2024 · 4c3beb5 · 4c3beb5
1 parent 3d46bf9
commit 4c3beb5
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 13 deletions.
diff --git a/model/KANamav1.py b/model/KANamav1.py
@@ -6,9 +6,9 @@
 import torch.nn as nn
 import torch.nn.functional as F
 
-from model.kan import KANLinear
-from model.args import ModelArgs
-from model.utils import RMSNorm, precompute_freqs_cis, apply_rotary_emb, repeat_kv
+from kan import KANLinear
+from args import ModelArgs
+from utils import RMSNorm, precompute_freqs_cis, apply_rotary_emb, repeat_kv
 
 
 class Attention(nn.Module):

diff --git a/model/KANamav2.py b/model/KANamav2.py
@@ -6,9 +6,9 @@
 import torch.nn as nn
 import torch.nn.functional as F
 
-from model.kan import KANLinear
-from model.args import ModelArgs
-from model.utils import RMSNorm, precompute_freqs_cis, apply_rotary_emb, repeat_kv
+from kan import KANLinear
+from args import ModelArgs
+from utils import RMSNorm, precompute_freqs_cis, apply_rotary_emb, repeat_kv
 
 
 class Attention(nn.Module):

diff --git a/model/KANamav3.py b/model/KANamav3.py
@@ -6,9 +6,9 @@
 import torch.nn as nn
 import torch.nn.functional as F
 
-from model.kan import KANLinear
-from model.args import ModelArgs
-from model.utils import RMSNorm, precompute_freqs_cis, apply_rotary_emb, repeat_kv
+from kan import KANLinear
+from args import ModelArgs
+from utils import RMSNorm, precompute_freqs_cis, apply_rotary_emb, repeat_kv
 
 
 class Attention(nn.Module):

diff --git a/model/KANamav4.py b/model/KANamav4.py
@@ -6,9 +6,9 @@
 import torch.nn as nn
 import torch.nn.functional as F
 
-from model.kan import KANLinear
-from model.args import ModelArgs
-from model.utils import RMSNorm, precompute_freqs_cis, apply_rotary_emb, repeat_kv
+from kan import KANLinear
+from args import ModelArgs
+from utils import RMSNorm, precompute_freqs_cis, apply_rotary_emb, repeat_kv
 
 
 class Attention(nn.Module):

diff --git a/setup.py b/setup.py
@@ -17,7 +17,7 @@
 # Setup configuration
 setup(
     name="KANama",
-    version="1.9.3",
+    version="1.9.4",
     description="KANama: marrying Kolmogorov–Arnold Networks with Meta's Llama model.",
     long_description=open(root_dir / "README.md", encoding="utf-8").read(),
     long_description_content_type="text/markdown",